add back rebalance in the finally block