Change the name from load_balancing to repartition