diff --git a/tests/shuffle/basic_shuffle.py b/tests/shuffle/basic_shuffle.py
index 898061c4ecf6a810f48b2ac6daf06a9926dd2530..6ba0e9d0099eda4a4e35f7eddfbf8a4cc2efb558 100644
--- a/tests/shuffle/basic_shuffle.py
+++ b/tests/shuffle/basic_shuffle.py
@@ -215,7 +215,7 @@ def basic_shuffle_2(conductor: ClusterConductor, dir, log: Logger):
 
         node_to_put = 0
         base_key = "key"
-        for i in range(0, 50):
+        for i in range(0, 300):
             r = c.put(node_to_put, f"{base_key}{i}", f"{i}", timeout=10)
             assert r.ok, f"expected ok for new key, got {r.status_code}"
             node_to_put += 1
@@ -243,7 +243,7 @@ def basic_shuffle_2(conductor: ClusterConductor, dir, log: Logger):
         for key in keys2:
             assert not (key in keys1)
         
-        assert original_get_all.keys() == keys1 + keys2
+        assert original_get_all.keys() == keys1 | keys2
         assert len(original_get_all) == len(keys1) + len(keys2)
 
         return True, "ok"