diff --git a/src/sugar/cli.py b/src/sugar/cli.py
index b958d06..a1115a0 100644
--- a/src/sugar/cli.py
+++ b/src/sugar/cli.py
@@ -6,7 +6,7 @@
 from pathlib import Path
 from typing import Tuple
 
-from core import Sugar
+from sugar.core import Sugar
 
 
 def _get_args():
diff --git a/src/sugar/core.py b/src/sugar/core.py
index f872bb3..b718ec4 100644
--- a/src/sugar/core.py
+++ b/src/sugar/core.py
@@ -13,7 +13,7 @@
 
 from jinja2 import Template
 
-from core import __version__
+from sugar import __version__
 from sugar.logs import KxgrErrorType, KxgrLogs
 
 
diff --git a/tests/test_containers_sugar.py b/tests/test_containers_sugar.py
index 1f99694..197181a 100644
--- a/tests/test_containers_sugar.py
+++ b/tests/test_containers_sugar.py
@@ -4,7 +4,7 @@
 
 import pytest
 
-from core import Sugar
+from sugar.core import Sugar
 
 CONFIG_PATH = Path(__file__).parent.parent / '.sugar.yaml'
 DEFAULT_ARGS = {