diff --git a/custom_components/ledfx/core/common.py b/custom_components/ledfx/core/common.py
index 7cae853..ab6abd3 100644
--- a/custom_components/ledfx/core/common.py
+++ b/custom_components/ledfx/core/common.py
@@ -5,6 +5,7 @@
 
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect, async_dispatcher_send
 from homeassistant.components.light import (
@@ -22,7 +23,6 @@
 from homeassistant.components.binary_sensor import BinarySensorEntity
 from homeassistant.components.sensor import SensorEntity
 from homeassistant.helpers.entity import Entity
-from homeassistant.const import ENTITY_CATEGORY_CONFIG, ENTITY_CATEGORY_DIAGNOSTIC
 
 from .const import DATA_UPDATED, DOMAIN, ICONS
 from .device import Device
@@ -174,7 +174,7 @@ def get_option(self, code: str):
 class LedFxSensor(SensorEntity, LedFxEntity):
     @property
     def entity_category(self) -> str:
-        return ENTITY_CATEGORY_DIAGNOSTIC
+        return EntityCategory.DIAGNOSTIC
 
     @property
     def native_value(self) -> Optional[str]:
@@ -183,7 +183,7 @@ def native_value(self) -> Optional[str]:
 class LedFxBinarySensor(BinarySensorEntity, LedFxEntity):
     @property
     def entity_category(self) -> str:
-        return ENTITY_CATEGORY_DIAGNOSTIC
+        return EntityCategory.DIAGNOSTIC
 
     @property
     def is_on(self) -> bool:
@@ -197,7 +197,7 @@ def is_on(self) -> bool:
 class LedFxSelect(SelectEntity, LedFxEntity):
     @property
     def entity_category(self) -> str:
-        return ENTITY_CATEGORY_CONFIG
+        return EntityCategory.CONFIG
 
     @property
     def current_option(self) -> Optional[str]:
@@ -321,7 +321,7 @@ def extra_state_attributes(self) -> dict:
 
     @property
     def entity_category(self) -> str:
-        return ENTITY_CATEGORY_CONFIG
+        return EntityCategory.CONFIG
 
     @property
     def available(self) -> bool:
diff --git a/custom_components/ledfx/manifest.json b/custom_components/ledfx/manifest.json
index 8b908f7..a12703f 100644
--- a/custom_components/ledfx/manifest.json
+++ b/custom_components/ledfx/manifest.json
@@ -1,7 +1,7 @@
 {
   "domain": "ledfx",
   "name": "LedFx",
-  "version": "1.3.1",
+  "version": "1.4.0",
   "documentation": "https://github.com/dmamontov/hass-ledfx",
   "issue_tracker": "https://github.com/dmamontov/hass-ledfx/issues",
   "config_flow": true,