diff --git a/pytest/test_sim_data_scalar.py b/pytest/test_sim_data_scalar.py
index 4105592..bb2209f 100644
--- a/pytest/test_sim_data_scalar.py
+++ b/pytest/test_sim_data_scalar.py
@@ -2,6 +2,6 @@
 
 def test_sim_data_scalar():
     scalar_survival_linear = sim_data_scalar(rho_G=0.25, rho_E=0.3, dim_G=500, dim_E=5, n=1500, dim_E_Sparse=2, ytype='Survival', n_inter=30)
-    assert scalar_survival_linear[0] is not None
-    assert scalar_survival_linear[1] is not None
+    assert scalar_survival_linear['data'] is not None
+    assert scalar_survival_linear['interpos'] is not None