use enum ConfigEntryDisabler

home-assistant · Dec 26, 2021 · 0e8e1e8 · 0e8e1e8
1 parent 746aa94
commit 0e8e1e8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/homeassistant/components/config/config_entries.py b/homeassistant/components/config/config_entries.py
@@ -311,6 +311,8 @@ async def config_entry_update(hass, connection, msg):
 async def config_entry_disable(hass, connection, msg):
     """Disable config entry."""
     disabled_by = msg["disabled_by"]
+    if disabled_by is not None:
+        disabled_by = config_entries.ConfigEntryDisabler(disabled_by)
 
     result = False
     try: