diff --git a/node_editor/node_editor.py b/node_editor/node_editor.py
index fcb5e438..a8ea140d 100644
--- a/node_editor/node_editor.py
+++ b/node_editor/node_editor.py
@@ -13,8 +13,8 @@
 import dearpygui.dearpygui as dpg
 from node.node_factory import NodeFactory
 import time
-from .style import STYLE
-from .util import _dpg_lock  # Import shared DearPyGUI lock
+from node_editor.style import STYLE
+from node_editor.util import _dpg_lock  # Import shared DearPyGUI lock
 from src.utils.logging import get_logger
 
 dpg.create_context()