Only remove update listener when unload was succesfull (#276)

Signed-off-by: Mick Vleeshouwer <[email protected]>
iMicknl · Oct 5, 2020 · 17d8a9b · 17d8a9b
1 parent 4ba6694
commit 17d8a9b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/custom_components/tahoma/__init__.py b/custom_components/tahoma/__init__.py
@@ -171,8 +171,6 @@ async def handle_execute_command(call):
 
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
     """Unload a config entry."""
-
-    hass.data[DOMAIN][entry.entry_id]["update_listener"]()
     entities_per_platform = hass.data[DOMAIN][entry.entry_id]["entities"]
 
     unload_ok = all(
@@ -185,6 +183,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
     )
 
     if unload_ok:
+        hass.data[DOMAIN][entry.entry_id]["update_listener"]()
         hass.data[DOMAIN].pop(entry.entry_id)
 
     return unload_ok