diff --git a/custom_components/mikrotik_router/binary_sensor.py b/custom_components/mikrotik_router/binary_sensor.py
index 24b068c..e05494e 100644
--- a/custom_components/mikrotik_router/binary_sensor.py
+++ b/custom_components/mikrotik_router/binary_sensor.py
@@ -3,9 +3,10 @@
 import logging
 from typing import Any, Dict, Optional
 
+
 from homeassistant.components.binary_sensor import (
     BinarySensorEntity,
-    DEVICE_CLASS_CONNECTIVITY,
+    BinarySensorDeviceClass,
 )
 from homeassistant.const import (
     CONF_NAME,
@@ -332,7 +333,7 @@ class MikrotikControllerPPPSecretBinarySensor(MikrotikControllerBinarySensor):
     @property
     def device_class(self) -> Optional[str]:
         """Return the device class."""
-        return DEVICE_CLASS_CONNECTIVITY
+        return BinarySensorDeviceClass.CONNECTIVITY
 
     @property
     def available(self) -> bool:
@@ -426,7 +427,7 @@ class MikrotikControllerPortBinarySensor(MikrotikControllerBinarySensor):
     @property
     def device_class(self) -> Optional[str]:
         """Return the device class."""
-        return DEVICE_CLASS_CONNECTIVITY
+        return BinarySensorDeviceClass.CONNECTIVITY
 
     @property
     def available(self) -> bool:
diff --git a/custom_components/mikrotik_router/sensor.py b/custom_components/mikrotik_router/sensor.py
index 943f29a..91180ef 100644
--- a/custom_components/mikrotik_router/sensor.py
+++ b/custom_components/mikrotik_router/sensor.py
@@ -7,12 +7,11 @@ from homeassistant.const import (
     CONF_NAME,
     ATTR_ATTRIBUTION,
     ATTR_DEVICE_CLASS,
-    DEVICE_CLASS_TEMPERATURE,
-    DEVICE_CLASS_TIMESTAMP,
-    DEVICE_CLASS_POWER,
     TEMP_CELSIUS,
 )
 
+from homeassistant.components.sensor import SensorDeviceClass
+
 from .const import (
     CONF_SENSOR_PORT_TRAFFIC,
     DEFAULT_SENSOR_PORT_TRAFFIC,
@@ -54,7 +53,7 @@ ATTR_ATTR = "data_attr"
 
 SENSOR_TYPES = {
     "system_temperature": {
-        ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
+        ATTR_DEVICE_CLASS: SensorDeviceClass.TEMPERATURE,
         ATTR_ICON: "mdi:thermometer",
         ATTR_LABEL: "Temperature",
         ATTR_UNIT: TEMP_CELSIUS,
@@ -63,7 +62,7 @@ SENSOR_TYPES = {
         ATTR_ATTR: "temperature",
     },
     "system_cpu-temperature": {
-        ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
+        ATTR_DEVICE_CLASS: SensorDeviceClass.TEMPERATURE,
         ATTR_ICON: "mdi:thermometer",
         ATTR_LABEL: "CPU temperature",
         ATTR_UNIT: TEMP_CELSIUS,
@@ -72,7 +71,7 @@ SENSOR_TYPES = {
         ATTR_ATTR: "cpu-temperature",
     },
     "system_board-temperature1": {
-        ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
+        ATTR_DEVICE_CLASS: SensorDeviceClass.TEMPERATURE,
         ATTR_ICON: "mdi:thermometer",
         ATTR_LABEL: "Board temperature",
         ATTR_UNIT: TEMP_CELSIUS,
@@ -81,7 +80,7 @@ SENSOR_TYPES = {
         ATTR_ATTR: "board-temperature1",
     },
     "system_power-consumption": {
-        ATTR_DEVICE_CLASS: DEVICE_CLASS_POWER,
+        ATTR_DEVICE_CLASS: SensorDeviceClass.POWER,
         ATTR_ICON: "mdi:transmission-tower",
         ATTR_LABEL: "Power consumption",
         ATTR_UNIT: "W",
@@ -108,7 +107,7 @@ SENSOR_TYPES = {
         ATTR_ATTR: "fan2-speed",
     },
     "system_uptime": {
-        ATTR_DEVICE_CLASS: DEVICE_CLASS_TIMESTAMP,
+        ATTR_DEVICE_CLASS: SensorDeviceClass.TIMESTAMP,
         ATTR_ICON: None,
         ATTR_LABEL: "Uptime",
         ATTR_UNIT: None,