Energy Dashboard Config - Individual Device not showing up for selection

Hi, totally stumped on this, having tried many combinations of device_class and state_class options, this sensor is not showing up in the Energy Dashboard config ‘individual device’ picker.

state_class: total_increasing
source: sensor.water_heater_kw
unit_of_measurement: kWh
device_class: energy
icon: mdi:chart-histogram
friendly_name: water_heater_kWh
last_reset: '2021-08-01T13:00:00'

configuraion.yaml

  - platform: integration
    source: sensor.water_heater_kw
    name: water_heater_kWh
    round: 2
    method: left

customize.yaml

sensor.water_heater_kWh:
  device_class: energy
  unit_of_measurement: kWh
  state_class: total_increasing
  last_reset: 2021-08-01T13:00:00

I’ve tried setting state_class to measurement, that didn’t work. Read quite a few posts here on similar topics, but still not found anything that can help.

Thank you.

Make sure you use the new template sensor. The old one does not support the features you need.