Unload cleanup

custom-components · Jul 22, 2024 · 5f9520b · 5f9520b
1 parent a5a1cf2
commit 5f9520b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/custom_components/healthchecksio/__init__.py b/custom_components/healthchecksio/__init__.py
@@ -59,10 +59,9 @@ async def async_unload_entry(
     hass: core.HomeAssistant, config_entry: config_entries.ConfigEntry
 ) -> bool:
     """Unload a config entry."""
-    unload_ok = await hass.config_entries.async_forward_entry_unload(
-        config_entry, Platform.BINARY_SENSOR
-    )
-    if unload_ok:
+    if unload_ok := await hass.config_entries.async_unload_platforms(
+        config_entry, PLATFORMS
+    ):
         hass.data.pop(DOMAIN_DATA, None)
         LOGGER.info("Successfully removed the healthchecksio integration")
     return unload_ok