Refactoring of PME load balancing