diff --git a/custom_components/husqvarna_automower/number.py b/custom_components/husqvarna_automower/number.py
index 2a1e58a..9ae1c0d 100644
--- a/custom_components/husqvarna_automower/number.py
+++ b/custom_components/husqvarna_automower/number.py
@@ -5,7 +5,7 @@
 from aiohttp import ClientResponseError
 from homeassistant.components.number import NumberEntity, NumberEntityDescription
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import TIME_MINUTES
+from homeassistant.const import UnitOfTime
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -45,14 +45,14 @@ async def async_setup_entry(
         translation_key="park_for",
         icon="mdi:clock-outline",
         entity_registry_enabled_default=True,
-        native_unit_of_measurement=TIME_MINUTES,
+        native_unit_of_measurement=UnitOfTime.MINUTES,
     ),
     NumberEntityDescription(
         key="Start",
         translation_key="mow_for",
         icon="mdi:clock-outline",
         entity_registry_enabled_default=True,
-        native_unit_of_measurement=TIME_MINUTES,
+        native_unit_of_measurement=UnitOfTime.MINUTES,
     ),
 )