diff --git a/OscCommentMacroListener.py b/OscCommentMacroListener.py
index 07d41bc..fea2bc7 100644
--- a/OscCommentMacroListener.py
+++ b/OscCommentMacroListener.py
@@ -16,8 +16,8 @@ __status__ = "Prototype"
 
 
 import logging
-from CommentMacroParser import CommentMacroParser
-from CommentMacroListener import CommentMacroListener
+from .CommentMacroParser import CommentMacroParser
+from .CommentMacroListener import CommentMacroListener
 from pythonosc import osc_message_builder
 from time import sleep
 
diff --git a/__init__.py b/__init__.py
index dd876c3..71c13d6 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1 +1 @@
-__all__ = ["comment"]
+__all__ = ["comment", "CommentMacroLexer", "CommentMacroListener", "CommentMacroParser", "OscCommentMacroListener"]
diff --git a/comment.py b/comment.py
index f6870d9..9a2025c 100755
--- a/comment.py
+++ b/comment.py
@@ -23,9 +23,9 @@ from antlr4 import CommonTokenStream
 from antlr4 import InputStream
 from antlr4 import ParseTreeWalker
 from antlr4.error.ErrorListener import ErrorListener
-from CommentMacroLexer import CommentMacroLexer
-from CommentMacroParser import CommentMacroParser
-from OscCommentMacroListener import OscCommentMacroListener
+from .CommentMacroLexer import CommentMacroLexer
+from .CommentMacroParser import CommentMacroParser
+from .OscCommentMacroListener import OscCommentMacroListener
 from pythonosc import udp_client