diff --git a/custom_components/openmower/entity.py b/custom_components/openmower/entity.py
index a866a78..3e2bbb8 100644
--- a/custom_components/openmower/entity.py
+++ b/custom_components/openmower/entity.py
@@ -4,13 +4,14 @@
 from typing import Optional
 
 from homeassistant.components import mqtt
-from homeassistant.components.openmower import DOMAIN
 from homeassistant.core import callback
 from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.entity import Entity
 from homeassistant.util import slugify, Throttle
 from homeassistant.util.json import json_loads_object
 
+from .const import DOMAIN
+
 MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=5)