diff --git a/homeassistant/components/xiaomi_aqara/__init__.py b/homeassistant/components/xiaomi_aqara/__init__.py
index 7a337dcc497..ae032a8b35f 100644
--- a/homeassistant/components/xiaomi_aqara/__init__.py
+++ b/homeassistant/components/xiaomi_aqara/__init__.py
@@ -1,9 +1,9 @@
 """Support for Xiaomi Gateways."""
+from datetime import timedelta
 import logging
 
-from datetime import timedelta
-
 import voluptuous as vol
+from xiaomi_gateway import XiaomiGatewayDiscovery
 
 from homeassistant.components.discovery import SERVICE_XIAOMI_GW
 from homeassistant.const import (
@@ -135,8 +135,6 @@ def setup(hass, config):
 
     discovery.listen(hass, SERVICE_XIAOMI_GW, xiaomi_gw_discovered)
 
-    from xiaomi_gateway import XiaomiGatewayDiscovery
-
     xiaomi = hass.data[PY_XIAOMI_GATEWAY] = XiaomiGatewayDiscovery(
         hass.add_job, gateways, interface
     )