diff --git a/tests/exec_test/check_model.py b/tests/exec_test/check_model.py
index 949f8a9c6f627f17f628d730e3c80ea097b71bd6..781759b8b976307ef2be20d201d43393e30ff138 100644
--- a/tests/exec_test/check_model.py
+++ b/tests/exec_test/check_model.py
@@ -1,9 +1,9 @@
-from cpp_sisso import Model
+from cpp_sisso import ModelRegressor
 from pathlib import Path
 
 import numpy as np
 
-model = Model(
+model = ModelRegressor(
     str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
 )