Fix rebalance test.