diff --git a/custom_components/ember_mug/sensor.py b/custom_components/ember_mug/sensor.py
index f6cfa40..b0d2983 100644
--- 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