diff --git a/tests/test_small_run.py b/tests/test_small_run.py
index 92c41eb..c77da0b 100644
--- a/tests/test_small_run.py
+++ b/tests/test_small_run.py
@@ -35,7 +35,7 @@ def _run(
         random.seed(10)
         ninduced = round(list_D_chunked[ksample]["X"].shape[0] * 0.35)
         D_tmp = list_D_chunked[ksample]
-        D[ksample]["Z"] = D_tmp["X"][random.sample(range(0, D_tmp["X"].shape[0] - 1), ninduced), :]
+        D_tmp[ksample]["Z"] = D_tmp["X"][random.sample(range(0, D_tmp["X"].shape[0] - 1), ninduced), :]
 
     # step 1 initialize model
     fit = process_multiSample.ini_multiSample(list_D_chunked, n_loadings, "nb", chol=False)