diff --git a/__init__.py b/__init__.py
index a378728..7432280 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1 +1,3 @@
-from nodes import *
+from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]