fix unload_entry

home-assistant · May 28, 2024 · cfdf377 · cfdf377
1 parent 6c59c80
commit cfdf377
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/homeassistant/components/point/__init__.py b/homeassistant/components/point/__init__.py
@@ -118,10 +118,10 @@ async def async_setup_webhook(
 async def async_unload_entry(hass: HomeAssistant, entry: PointConfigEntry) -> bool:
     """Unload a config entry."""
     if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
-        webhook.async_unregister(hass, entry.data[CONF_WEBHOOK_ID])
-        session: PointSession = hass.data[DOMAIN].pop(entry)
-        await session.remove_webhook()
-
+        session: PointSession = hass.data[DOMAIN].pop(entry.entry_id)
+        if CONF_WEBHOOK_ID in entry.data:
+            webhook.async_unregister(hass, entry.data[CONF_WEBHOOK_ID])
+            await session.remove_webhook()
     return unload_ok