diff --git a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py
index 9e43ca60f74b..83f094df2aac 100644
--- a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py
+++ b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py
@@ -38,8 +38,8 @@ def setUpClass(cls):
                 128,
                 "--enable-piecewise-cuda-graph",
                 "--piecewise-cuda-graph-tokens",
-                TOKENS_TO_CAPTURE,
-            ],
+            ]
+            + TOKENS_TO_CAPTURE,
         )
 
     @classmethod
@@ -79,8 +79,8 @@ def test_latency(self):
                 "ascend",
                 "--enable-piecewise-cuda-graph",
                 "--piecewise-cuda-graph-tokens",
-                TOKENS_TO_CAPTURE,
-            ],
+            ]
+            + TOKENS_TO_CAPTURE,
         )
         self.assertLess(prefill_latency, EXP_PREFILL_LATENCY)