Schedule Unifi shutdown callback earlier (home-assistant#33257)

jlvaillant · Mar 25, 2020 · 5650b39 · 5650b39
1 parent 8f4d314
commit 5650b39
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/homeassistant/components/unifi/__init__.py b/homeassistant/components/unifi/__init__.py
@@ -40,6 +40,8 @@ async def async_setup_entry(hass, config_entry):
     controller_id = get_controller_id_from_config_entry(config_entry)
     hass.data[DOMAIN][controller_id] = controller
 
+    hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, controller.shutdown)
+
     if controller.mac is None:
         return True
 
@@ -53,8 +55,6 @@ async def async_setup_entry(hass, config_entry):
         # sw_version=config.raw['swversion'],
     )
 
-    hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, controller.shutdown)
-
     return True