diff --git a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py
index 36d874d02..8ce22bedd 100644
--- a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py
+++ b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py
@@ -1222,7 +1222,11 @@ class TestMlxOverlapScheduler(unittest.TestCase):
                 set_completion_time=lambda: events.append(("completion", "r0"))
             ),
         )
-        batch = SimpleNamespace()
+        batch = SimpleNamespace(
+            mamba_track_mask_cpu=None,
+            mamba_track_mask_next_cpu=None,
+            mamba_decode_batch_idx_cpu=None,
+        )
         result = SimpleNamespace()
         i = 0
         logits_output = SimpleNamespace(customized_info=None)