Move rebalance call into delete method