diff --git a/custom_components/openwbmqtt/number.py b/custom_components/openwbmqtt/number.py
index f84d9ce..f7c890f 100644
--- a/custom_components/openwbmqtt/number.py
+++ b/custom_components/openwbmqtt/number.py
@@ -12,9 +12,9 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
     DEVICE_DEFAULT_NAME,
-    ELECTRIC_CURRENT_AMPERE,
-    ENERGY_KILO_WATT_HOUR,
-    ENTITY_CATEGORY_CONFIG,
+    UnitOfElectricCurrent,
+    UnitOfEnergy,
+    EntityCategory,
     PERCENTAGE,
 )
 from homeassistant.core import HomeAssistant, callback