We consider a control problem for a diffusive PDE model of heterogeneous population of thermostatically controlled loads (TCLs) aiming to balance the aggregate power consumption within a given amount of time. Using the Green’s function approach, the problem is formulated as an approximate controllability problem for a residue depending on control parameters nonlinearly. A sufficient condition for approximate controllability is derived in terms of initial temperature distribution, operation time of TCLs and threshold value of the aggregate power consumption. Case studies allow to reveal the advantages of the proposed solution from numerical calculations point of view.