diff --git a/tests/bench/benchmark.py b/tests/bench/benchmark.py
index ed17ee77fd46b7201874f6f66a78cf07150b006d..97ef0d14da2849c94caa7a22e6d053538e7e0fd6 100644
--- a/tests/bench/benchmark.py
+++ b/tests/bench/benchmark.py
@@ -9,6 +9,7 @@ import matplotlib.pyplot as plt
 
 NUM_SHARDS = 16
 NUM_KEYS = 10000
+
 def benchmark_add_shard(conductor: ClusterConductor, dir, log: Logger):
     with KVSTestFixture(conductor, dir, log, node_count=NUM_SHARDS) as fx:
         conductor.add_shard("shard1", conductor.get_nodes([0]))
@@ -16,7 +17,7 @@ def benchmark_add_shard(conductor: ClusterConductor, dir, log: Logger):
 
         log("putting 100 keys\n")
         put_times = []
-        for i in range(NUM_SHARDS):
+        for i in range(NUM_KEYS):
             c = KVSMultiClient(fx.clients, "client", log)
             start_time = time.time()
             r = c.put(0, f"key{i}", f"value{i}", timeout=10)