There is provided a method of controlling aerosol production in an aerosol-generating device, the device comprising: a heater comprising at least one heating element configured to heat an aerosol-forming substrate; and a power source for providing power to the heating element, comprising the steps of: controlling the power provided to the heating element such that in a first phase power is provided such that the temperature of the heating element increases from an initial temperature to a first temperature, in a second phase power is provided such that the temperature of the heating element drops below the first temperature and in a third phase power is provided such that the temperature of the heating element increases again. Increasing the temperature of the heating element during a final phase of the heating process reduces or prevents the reduction in aerosol delivery over time.