diff --git a/homeassistant/components/config/device_registry.py b/homeassistant/components/config/device_registry.py index 50d56915dd4..5e7c2ef1938 100644 --- a/homeassistant/components/config/device_registry.py +++ b/homeassistant/components/config/device_registry.py @@ -62,6 +62,9 @@ async def websocket_update_device(hass, connection, msg): msg.pop("type") msg_id = msg.pop("id") + if "disabled_by" in msg: + msg["disabled_by"] = DeviceEntryDisabler(msg["disabled_by"]) + entry = registry.async_update_device(**msg) connection.send_message(websocket_api.result_message(msg_id, _entry_dict(entry)))