fix: attribute name error

sopelj · Dec 26, 2023 · 59bd30a · 59bd30a
1 parent 0490795
commit 59bd30a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/ember_mug/sensor.py b/custom_components/ember_mug/sensor.py
@@ -126,7 +126,7 @@ class EmberMugLiquidLevelSensor(EmberMugSensor):
     @cached_property
     def max_level(self) -> int:
         """Max level is different for travel mug."""
-        if self.coordinator.mug.data.model_info.type == DeviceType.TRAVEL_MUG:
+        if self.coordinator.mug.data.model_info.device_type == DeviceType.TRAVEL_MUG:
             return 100
         return 30