diff --git a/homeassistant/components/fan/xiaomi_miio.py b/homeassistant/components/fan/xiaomi_miio.py
index 2538e8fcd1f..264962b9d56 100644
--- a/homeassistant/components/fan/xiaomi_miio.py
+++ b/homeassistant/components/fan/xiaomi_miio.py
@@ -302,7 +302,7 @@ class XiaomiAirPurifier(FanEntity):
 
         yield from self._try_command(
             "Setting operation mode of the air purifier failed.",
-            self._air_purifier.set_mode, OperationMode[speed])
+            self._air_purifier.set_mode, OperationMode(speed))
 
     @asyncio.coroutine
     def async_set_buzzer_on(self):