diff --git a/tests/test_SequentialDesign.py b/tests/test_SequentialDesign.py
index 15b0a4c494bebdc8893d04bacefd48ec217dae53..7a9bf228c27a9870041b53c08c84d21f3e25c135 100644
--- a/tests/test_SequentialDesign.py
+++ b/tests/test_SequentialDesign.py
@@ -166,7 +166,7 @@ def test_choose_next_sample() -> None:
 
     mm = PCE(inp)
     mm.fit(expdes.X, expdes.Y)
-    expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+    expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
     mod = PL()
     
     engine = Engine(mm, mod, expdes)
@@ -196,7 +196,7 @@ def test_choose_next_sample_da_spaceparallel() -> None:
 
     mm = PCE(inp)
     mm.fit(expdes.X, expdes.Y)
-    expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+    expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
     mod = PL()
     
     engine = Engine(mm, mod, expdes)
@@ -227,7 +227,7 @@ def test_choose_next_sample_da_spacenoparallel() -> None:
 
     mm = PCE(inp)
     mm.fit(expdes.X, expdes.Y)
-    expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+    expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
     mod = PL()
     
     engine = Engine(mm, mod, expdes)
@@ -258,7 +258,7 @@ def test_choose_next_sample_loo_space() -> None:
 
     mm = PCE(inp)
     mm.fit(expdes.X, expdes.Y)
-    expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+    expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
     mod = PL()
     
     engine = Engine(mm, mod, expdes)
@@ -288,7 +288,7 @@ def test_choose_next_sample_vor_space() -> None:
 
     mm = PCE(inp)
     mm.fit(expdes.X, expdes.Y)
-    expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+    expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
     mod = PL()
     
     engine = Engine(mm, mod, expdes)
@@ -323,7 +323,7 @@ def test_choose_next_sample_vor_space() -> None:
 
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -353,7 +353,7 @@ def test_choose_next_sample_vor_space() -> None:
 
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -383,7 +383,7 @@ def test_choose_next_sample_vor_space() -> None:
 
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -413,7 +413,7 @@ def test_choose_next_sample_vor_space() -> None:
 
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -444,7 +444,7 @@ def test_choose_next_sample_vor_space() -> None:
 
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -475,7 +475,7 @@ def test_choose_next_sample_vor_space() -> None:
 
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -508,7 +508,7 @@ def test_choose_next_sample_vor_space() -> None:
     
 #         mm = PCE(inp)
 #         mm.fit(expdes.X, expdes.Y)
-#         expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#         expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #         mod = PL()
         
 #         engine = Engine(mm, mod, expdes)
@@ -539,7 +539,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'MI'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -572,7 +572,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'MI'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -610,7 +610,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'ALC'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -643,7 +643,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'DKL'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -676,7 +676,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'DPP'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -709,7 +709,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'APP'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -742,7 +742,7 @@ def test_choose_next_sample_vor_space() -> None:
 #     expdes.util_func = 'MI'
 #     mm = PCE(inp)
 #     mm.fit(expdes.X, expdes.Y)
-#     expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#     expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #     mod = PL()
     
 #     engine = Engine(mm, mod, expdes)
@@ -775,7 +775,7 @@ def test_choose_next_sample_vor_space() -> None:
 #         expdes.util_func = 'BME'
 #         mm = PCE(inp)
 #         mm.fit(expdes.X, expdes.Y)
-#         expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#         expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #         mod = PL()
         
 #         engine = Engine(mm, mod, expdes)
@@ -809,7 +809,7 @@ def test_choose_next_sample_vor_space() -> None:
 #         expdes.util_func = 'DKL'
 #         mm = PCE(inp)
 #         mm.fit(expdes.X, expdes.Y)
-#         expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#         expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #         mod = PL()
         
 #         engine = Engine(mm, mod, expdes)
@@ -843,7 +843,7 @@ def test_choose_next_sample_vor_space() -> None:
 #         expdes.util_func = 'infEntropy'
 #         mm = PCE(inp)
 #         mm.fit(expdes.X, expdes.Y)
-#         expdes.generate_ED(expdes.n_init_samples, max_pce_deg=np.max(mm._pce_deg))
+#         expdes.generate_ED(max_pce_deg=np.max(mm._pce_deg))
 #         mod = PL()
         
 #         engine = Engine(mm, mod, expdes)