diff --git a/custom_components/easy_computer_manage/__init__.py b/custom_components/easy_computer_manage/__init__.py
index 37f56e7..1424bea 100644
--- a/custom_components/easy_computer_manage/__init__.py
+++ b/custom_components/easy_computer_manage/__init__.py
@@ -67,3 +67,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     )
 
     return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Unload a config entry."""
+    return await hass.config_entries.async_forward_entry_unload(
+        entry, "switch"
+    )