diff --git a/custom_components/melcloud_custom/sensor.py b/custom_components/melcloud_custom/sensor.py
index 704fd6f..5433591 100644
--- a/custom_components/melcloud_custom/sensor.py
+++ b/custom_components/melcloud_custom/sensor.py
@@ -16,7 +16,11 @@
     SensorStateClass,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import UnitOfEnergy, UnitOfTemperature
+from homeassistant.const import (
+    UnitOfEnergy,
+    UnitOfTemperature,
+    SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+)
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback