diff --git a/rebalance-zone-expansion.sh b/rebalance-zone-expansion.sh
index 938dcfedc4..51e4a356b0 100755
--- a/rebalance-zone-expansion.sh
+++ b/rebalance-zone-expansion.sh
@@ -126,6 +126,7 @@ $vold_home/bin/run-class.sh voldemort.tools.RepartitionerCLI \
                             --current-stores $final_stores \
                             --output-dir $output_dir/step2 \
                             --enable-random-swaps \
+                            --random-swap-zoneids 2 \
                             --attempts 5 \
                             --random-swap-attempts 1000 \