diff --git a/test/registered/attention/test_aiter_fp8_q_unified_attention.py b/test/registered/attention/test_aiter_fp8_q_unified_attention.py
index bc1c608f8..88c2dfb74 100644
--- a/test/registered/attention/test_aiter_fp8_q_unified_attention.py
+++ b/test/registered/attention/test_aiter_fp8_q_unified_attention.py
@@ -59,6 +59,7 @@ class TestAiterFP8QUnifiedAttention(CustomTestCase):
         backend.use_mla = branch == "mla"
         backend.kv_cache_is_vectorized_5d = branch == "vectorized"
         backend.use_triton_unified_attention = branch != "legacy"
+        backend.dcp_world_size = 1
         backend.kv_cache_dtype = kv_cache_dtype
         backend.input_dtype = torch.bfloat16
         backend.page_size = 1