diff --git a/tests/test_small_run.py b/tests/test_small_run.py
index ddd7c42..eb5a93e 100644
--- a/tests/test_small_run.py
+++ b/tests/test_small_run.py
@@ -80,7 +80,7 @@ def run_cli(
     data_dir: str | Path,
     output_dir: str | Path,
     n_loadings: int = 1,
-    n_sample: int = 1,
+    n_sample: int = 2,
     epochs: int = 10,
     legacy: bool = True,
 ):