using retry for rebalance and resolving some issues