Minor change to rebalancing