diff --git a/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py b/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py
index 40fe6656c..ac8176210 100644
--- a/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py
+++ b/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py
@@ -15,8 +15,8 @@ from sglang.test.test_utils import (
 register_cuda_ci(est_time=160, suite="stage-b-test-large-2-gpu")
 
 
-class TestPiecewiseCudaGraphFusedMoE(CustomTestCase):
-    """Test piecewise CUDA graph with FusedMoE Backend"""
+class TestPiecewiseCudaGraphTP(CustomTestCase):
+    """Test piecewise CUDA graph with normal TP"""
 
     @classmethod
     def setUpClass(cls):
@@ -32,8 +32,6 @@ class TestPiecewiseCudaGraphFusedMoE(CustomTestCase):
                 "eager",
                 "--tp",
                 "2",
-                "--ep-size",
-                "2",
             ],
         )