diff --git a/fastvideo/tests/training/self-forcing/test_self_forcing.py b/fastvideo/tests/training/self-forcing/test_self_forcing.py
index ebfa2c145..1ab4f5c7c 100644
--- a/fastvideo/tests/training/self-forcing/test_self_forcing.py
+++ b/fastvideo/tests/training/self-forcing/test_self_forcing.py
@@ -37,7 +37,7 @@ def run_worker():
         "--data_path", "data/crush-smol_processed_t2v/combined_parquet_dataset",
         "--validation_dataset_file", "examples/training/finetune/wan_t2v_1.3B/crush_smol/validation.json",
         "--train_batch_size", "1",
-        "--num_latent_t", "6", 
+        "--num_latent_t", "21", 
         "--num_gpus", "2",
         "--sp_size", "1",
         "--tp_size", "1",
@@ -62,7 +62,7 @@ def run_worker():
         "--wandb_run_name", wandb_name,
         "--num_height", "480",
         "--num_width", "832",
-        "--num_frames", "6",
+        "--num_frames", "21",
         "--flow_shift", "5",
         "--validation_guidance_scale", "1.0",
         "--weight_decay", "0.01",
@@ -83,7 +83,7 @@ def run_worker():
         "--simulate_generator_forward",
         "--num_frame_per_block", "3",
         "--enable_gradient_masking",
-        "--gradient_mask_last_n_frames", "6",
+        "--gradient_mask_last_n_frames", "21",
         "--independent_first_frame", "False",
         "--same_step_across_blocks", "True",
         "--last_step_only", "False",