diff --git a/__init__.py b/__init__.py
index 48208d6..fe724f9 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,5 @@
 """Top-level package for basic_data_handling."""
+import os, sys; sys.path.append(os.path.dirname(os.path.realpath(__file__)))
 
 __all__ = [
     "NODE_CLASS_MAPPINGS",
@@ -6,7 +7,7 @@ __all__ = [
     "WEB_DIRECTORY",
 ]
 
-from .src.basic_data_handling import NODE_CLASS_MAPPINGS
-from .src.basic_data_handling import NODE_DISPLAY_NAME_MAPPINGS
+from src.basic_data_handling import NODE_CLASS_MAPPINGS
+from src.basic_data_handling import NODE_DISPLAY_NAME_MAPPINGS
 
 WEB_DIRECTORY = "./web"