fix rebalance