diff --git a/drawbridge/__init__.py b/drawbridge/__init__.py
index e05180e..389584f 100644
--- a/drawbridge/__init__.py
+++ b/drawbridge/__init__.py
@@ -2,6 +2,7 @@ import os
 
 import iptc
 
+from .drawbridge import DrawBridge
 from .net_queue import NetQueue
 
 
diff --git a/drawbridge/drawbridge.py b/drawbridge/drawbridge.py
index c68e585..3e468fb 100644
--- a/drawbridge/drawbridge.py
+++ b/drawbridge/drawbridge.py
@@ -5,8 +5,8 @@ from typing import Optional
 
 import fnfqueue
 
-from net_queue import NetQueue
-from utils.logger import logger
+from .net_queue import NetQueue
+from .utils.logger import logger
 
 
 class DrawBridge: