diff --git a/tests/shuffle/basic_shuffle.py b/tests/shuffle/basic_shuffle.py
index c23cda659d3788ff92a33921dfb0982722612120..68c98ed5669d85a1a19e8b886f3cd77ea016216d 100644
--- a/tests/shuffle/basic_shuffle.py
+++ b/tests/shuffle/basic_shuffle.py
@@ -72,7 +72,8 @@ def basic_shuffle(conductor: ClusterConductor, dir, log: Logger):
 
         # Remove shard 3, causing a shuffle. Move Node 2 to shard 1 so the keys should still exist, and be shuffled
         conductor.remove_shard("shard3")
-        conductor.add_node_to_shard("shard1", conductor.get_nodes([2]))
+        conductor.add_node_to_shard("shard1", conductor.get_node(2))
+        fx.broadcast_view(conductor.get_shard_view())
 
         r = c.get_all(0, timeout=10) # should get all of shard 1's keys
         assert r.ok, f"expected ok for get, got {r.status_code}"