rebalance fix