diff --git a/tests/distributed/test_eplb_execute.py b/tests/distributed/test_eplb_execute.py
index dfb15311202e5..14ad148da6c45 100644
--- a/tests/distributed/test_eplb_execute.py
+++ b/tests/distributed/test_eplb_execute.py
@@ -312,7 +312,7 @@ def _test_async_transfer_layer_without_mtp_worker(
             is_received_locally=is_received_locally,
             recv_metadata=recv_metadata,
             new_indices=new_indices_cpu[layer_idx],
-            ep_group=ep_group,
+            ep_rank=ep_group.rank(),
         )
 
     verify_expert_weights_after_shuffle(