A removable heater assembly for an aerosol-generating device is provided, the removable heater assembly being configured to heat a tobacco plug and including an electrical heater, a data storage device, and data stored on the data storage device, the data including calibration data for the electrical heater. An aerosol-generating device is also provided, including the removable heater assembly, a housing defining a cavity configured to removably receive the removable heater assembly and an aerosol-generating article, a power supply, and a controller configured to receive data from the data storage device of the removable heater assembly and to control a supply of electrical power from the power supply to the electrical heater of the removable heater assembly based on the received data.