The balance factor was not updated after the partition id was assigned to an edge. This should fix load balancing problem.