diff --git a/test/registered/attention/test_mamba_unittest.py b/test/registered/attention/test_mamba_unittest.py
index 0a0f45fb8..937c16baa 100644
--- a/test/registered/attention/test_mamba_unittest.py
+++ b/test/registered/attention/test_mamba_unittest.py
@@ -20,7 +20,10 @@ register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu")
 class TestMamba(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        pass
+        server_args = ServerArgs(model_path="dummy")
+        server_args.enable_dp_attention = False
+        server_args.page_size = 1
+        set_global_server_args_for_scheduler(server_args)
 
     @classmethod
     def tearDownClass(cls):