diff --git a/tests/shuffle/basic_shuffle.py b/tests/shuffle/basic_shuffle.py
index a8fbd5f1a75680b922eb5164386632d3647a103e..c23cda659d3788ff92a33921dfb0982722612120 100644
--- a/tests/shuffle/basic_shuffle.py
+++ b/tests/shuffle/basic_shuffle.py
@@ -86,15 +86,18 @@ def basic_shuffle(conductor: ClusterConductor, dir, log: Logger):
 
         assert len(shard1_keys) + len(shard2_keys) == 15, f"expected 15 keys, got {len(shard1_keys) + len(shard2_keys)}"
 
+        # Remove shard 2. This loses keys.
+        conductor.remove_shard("shard2")
         fx.broadcast_view(conductor.get_shard_view())
 
         r = c.get_all(0, timeout=10)
         assert r.ok, f"expected ok for get, got {r.status_code}"
         res = r.json()["items"]
         shard1_keys_after_delete = res
-
+        
         assert len(shard1_keys) == len(shard1_keys_after_delete), f"expected {len(shard1_keys)} keys, got {len(shard1_keys_after_delete)}"
 
+
         return True, "ok"
     
 def partitioned_shards(conductor: ClusterConductor, dir, log: Logger):