diff --git a/custom_components/mikrotik_router/__init__.py b/custom_components/mikrotik_router/__init__.py
index 30a13a9..064e964 100644
--- a/custom_components/mikrotik_router/__init__.py
+++ b/custom_components/mikrotik_router/__init__.py
@@ -11,7 +11,7 @@ from homeassistant.const import (
 	CONF_SSL,
 )
 
-from mikrotik_controller import MikrotikControllerData
+from .mikrotik_controller import MikrotikControllerData
 
 from .const import (
 	DOMAIN,
diff --git a/custom_components/mikrotik_router/config_flow.py b/custom_components/mikrotik_router/config_flow.py
index 718963f..1be02f2 100644
--- a/custom_components/mikrotik_router/config_flow.py
+++ b/custom_components/mikrotik_router/config_flow.py
@@ -15,7 +15,7 @@ from homeassistant.const import (
 	CONF_SSL,
 )
 
-from const import (
+from .const import (
 	DOMAIN,
 	CONF_TRACK_ARP,
 	DEFAULT_TRACK_ARP,