diff --git a/homeassistant/components/demo/update.py b/homeassistant/components/demo/update.py
index 648ad59bb55..15e67ffa0a8 100644
--- a/homeassistant/components/demo/update.py
+++ b/homeassistant/components/demo/update.py
@@ -4,8 +4,11 @@ from __future__ import annotations
 import asyncio
 from typing import Any
 
-from homeassistant.components.update import UpdateDeviceClass, UpdateEntity
-from homeassistant.components.update.const import UpdateEntityFeature
+from homeassistant.components.update import (
+    UpdateDeviceClass,
+    UpdateEntity,
+    UpdateEntityFeature,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import DEVICE_DEFAULT_NAME
 from homeassistant.core import HomeAssistant
diff --git a/homeassistant/components/zwave_js/update.py b/homeassistant/components/zwave_js/update.py
index 932ed46a0fc..022c4dc3f3c 100644
--- a/homeassistant/components/zwave_js/update.py
+++ b/homeassistant/components/zwave_js/update.py
@@ -20,8 +20,11 @@ from zwave_js_server.model.firmware import (
 )
 from zwave_js_server.model.node import Node as ZwaveNode
 
-from homeassistant.components.update import UpdateDeviceClass, UpdateEntity
-from homeassistant.components.update.const import UpdateEntityFeature
+from homeassistant.components.update import (
+    UpdateDeviceClass,
+    UpdateEntity,
+    UpdateEntityFeature,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.exceptions import HomeAssistantError