diff --git a/custom_components/raincloud/__init__.py b/custom_components/raincloud/__init__.py
index 9d6b0f7..168ef0c 100644
--- a/custom_components/raincloud/__init__.py
+++ b/custom_components/raincloud/__init__.py
@@ -14,7 +14,7 @@
     CONF_USERNAME,
     TIME_DAYS,
     TIME_MINUTES,
-    UNIT_PERCENTAGE,
+    PERCENTAGE,
 )
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.dispatcher import async_dispatcher_connect, dispatcher_send
@@ -60,7 +60,7 @@
 
 UNIT_OF_MEASUREMENT_MAP = {
     "auto_watering": "",
-    "battery": UNIT_PERCENTAGE,
+    "battery": PERCENTAGE,
     "is_watering": "",
     "manual_watering": "",
     "next_cycle": "",