diff --git a/tests/test_pipeline_real_weights.py b/tests/test_pipeline_real_weights.py
index 2e2f2573..570c6ac0 100644
--- a/tests/test_pipeline_real_weights.py
+++ b/tests/test_pipeline_real_weights.py
@@ -37,7 +37,7 @@ DEVICE = "cuda"
 
 # Stage control (1-6)
 STAGE_START = 1
-STAGE_END = 6
+STAGE_END = 2
 
 # ============================================================
 # Weight loading