{{states('water_heater.smart_boiler_9731')}} results in Unknown, while {{states('sensor.smart_boiler_9731_s_heater')}} results in the correct status. Is it possible to display the correct status in the water_heater entity?