fixing rebalancing