diff --git a/tests/test_small_run.py b/tests/test_small_run.py
index e9f82d2..6efe10e 100644
--- a/tests/test_small_run.py
+++ b/tests/test_small_run.py
@@ -37,7 +37,7 @@ def _run(
         D[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(D, n_loadings)
+    fit = process_multiSample.ini_multiSample(D, n_loadings, "nb")
 
     # step 2 fit model