diff --git a/custom_components/xiaomi_viomi/vacuum.py b/custom_components/xiaomi_viomi/vacuum.py
index 92364db..9a70d69 100644
--- a/custom_components/xiaomi_viomi/vacuum.py
+++ b/custom_components/xiaomi_viomi/vacuum.py
@@ -230,7 +230,7 @@ def update(self):
             state = self._get_device_status()
             self.vacuum_state = state
 
-            self._fan_speeds = state.fan_speed_presets()
+            self._fan_speeds = {x.name: x.value for x in list(ViomiVacuumSpeed)}
             self._fan_speeds_reverse = {v: k for k, v in self._fan_speeds.items()}
 
             self.consumable_state = self._device.consumable_status()