diff --git a/tests/exec_test/classification_gen_proj/check_model.py b/tests/exec_test/classification_gen_proj/check_model.py
index 5958c75a5d751910b4baa9cd22593a2de8289c5d..c004eb5d59bfce0ef5189bd59e10b8e96a12e3a3 100644
--- a/tests/exec_test/classification_gen_proj/check_model.py
+++ b/tests/exec_test/classification_gen_proj/check_model.py
@@ -4,7 +4,7 @@ from pathlib import Path
 import numpy as np
 
 model = ModelClassifier(
-    str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
+    str("models/train_dim_2_model_0.dat")
 )
 assert model.percent_error < 1e-7
 assert model.percent_test_error < 1e-7
diff --git a/tests/exec_test/gen_proj/check_model.py b/tests/exec_test/gen_proj/check_model.py
index 183a6c1712ceacec2a96d313d41a9f191952bb0d..640d2b9aa45cdd8092c75a3b7858a55c4150af84 100644
--- a/tests/exec_test/gen_proj/check_model.py
+++ b/tests/exec_test/gen_proj/check_model.py
@@ -4,7 +4,7 @@ from pathlib import Path
 import numpy as np
 
 model = ModelRegressor(
-    str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
+    str("models/train_dim_2_model_0.dat")
 )
 assert model.rmse < 1e-4
 assert model.test_rmse < 1e-4
diff --git a/tests/exec_test/log_reg_gen_proj/check_model.py b/tests/exec_test/log_reg_gen_proj/check_model.py
index c5cdbe83019b9eaea840c312ad60630f00a84896..311535954b81340d69652add406418539dcfd65f 100644
--- a/tests/exec_test/log_reg_gen_proj/check_model.py
+++ b/tests/exec_test/log_reg_gen_proj/check_model.py
@@ -4,7 +4,7 @@ from pathlib import Path
 import numpy as np
 
 model = ModelLogRegressor(
-    str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
+    str("models/train_dim_2_model_0.dat")
 )
 assert model.rmse < 1e-4
 assert model.test_rmse < 1e-4