diff --git a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py
index 83f094df2aac..7f13d13e8bb4 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