diff --git a/binary_sensor.py b/binary_sensor.py
index f14d547..cd7022f 100644
--- a/binary_sensor.py
+++ b/binary_sensor.py
@@ -7,7 +7,7 @@ from homeassistant.components.binary_sensor import (
 )
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from ha_rdz_pdc_config import DOMAIN, HeatPumpDataCoordinator
+from . import DOMAIN, HeatPumpDataCoordinator
 
 async def async_setup_entry(hass, entry, async_add_entities):
     """Set up binary sensors."""
diff --git a/sensor.py b/sensor.py
index b545285..caa614b 100644
--- a/sensor.py
+++ b/sensor.py
@@ -9,7 +9,7 @@ from homeassistant.components.sensor import (
 from homeassistant.const import UnitOfTemperature, UnitOfTime
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from ha_rdz_pdc_config import DOMAIN, HeatPumpDataCoordinator
+from . import DOMAIN, HeatPumpDataCoordinator
 
 async def async_setup_entry(hass, entry, async_add_entities):
     """Set up sensors."""