diff --git a/pytest/test_grid_scalar_ge.py b/pytest/test_grid_scalar_ge.py
index f541f34..1914d3b 100644
--- a/pytest/test_grid_scalar_ge.py
+++ b/pytest/test_grid_scalar_ge.py
@@ -22,6 +22,6 @@ def test_grid_scalar_ge():
                                         Learning_Rate2=[0.035,0.045], L2=[0.1], Learning_Rate1=[0.04, 0.05, 0.06], L=[0.08, 0.09], 
                                         Num_Epochs=50, t = 0.01, model = None, split_type = 1, ratio = [7, 2, 1], 
                                         important_feature = False, plot = False, model_reg = None, isfunc = False)
-  assert scalar_ge_res_1 is not None
-  assert scalar_ge_res_2 is not None
-  assert scalar_ge_res_3 is not None
+  assert grid_scalar_ge_res_1 is not None
+  assert grid_scalar_ge_res_2 is not None
+  assert grid_scalar_ge_res_3 is not None