diff --git a/flowslicer.py b/flowslicer.py
index 5eaaa38..26dddd8 100644
--- a/flowslicer.py
+++ b/flowslicer.py
@@ -16,8 +16,12 @@
 
 from collections.abc import Mapping
 
-import db
-from dfil import DataNode, DataFlowEdge, TokenExpression
+try:
+    import db
+    from dfil import DataNode, DataFlowEdge, TokenExpression
+except ImportError:
+    import flowslicer.db as db
+    from flowslicer.dfil import DataNode, DataFlowEdge, TokenExpression
 
 try:
     from .dfil import *