diff --git a/fastvideo/tests/training/self-forcing/test_self_forcing.py b/fastvideo/tests/training/self-forcing/test_self_forcing.py
index ad130e4b5..2a3229c19 100644
--- a/fastvideo/tests/training/self-forcing/test_self_forcing.py
+++ b/fastvideo/tests/training/self-forcing/test_self_forcing.py
@@ -39,7 +39,7 @@ def run_worker():
         "--train_batch_size", "1",
         "--num_latent_t", "6", 
         "--num_gpus", "2",
-        "--sp_size", "2",
+        "--sp_size", "1",
         "--tp_size", "1",
         "--hsdp_replicate_dim", "1",
         "--hsdp_shard_dim", "2",