Controlling cluster rebalancing