fix: icon property

sopelj · Jan 15, 2022 · f167b76 · f167b76
1 parent 8188daf
commit f167b76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/custom_components/ember_mug/sensor.py b/custom_components/ember_mug/sensor.py
@@ -71,6 +71,7 @@ def available(self) -> bool:
 class EmberMugSensor(EmberMugSensorBase, SensorEntity):
     """Base Mug State Sensor."""
 
+    @property
     def icon(self) -> str | None:
         """Change icon based on state."""
         return ICON_EMPTY if self.state == "Empty" else ICON_DEFAULT