diff --git a/src/kapidox/depdiagram/__init__.py b/src/kapidox/depdiagram/__init__.py
--- a/src/kapidox/depdiagram/__init__.py
+++ b/src/kapidox/depdiagram/__init__.py
@@ -1 +1,5 @@
-from kapidox.depdiagram.generate import *
+import sys
+if sys.version_info.major < 3:
+    from generate import *
+else:
+    from kapidox.depdiagram.generate import *
diff --git a/src/kapidox/depdiagram/frameworkdb.py b/src/kapidox/depdiagram/frameworkdb.py
--- a/src/kapidox/depdiagram/frameworkdb.py
+++ b/src/kapidox/depdiagram/frameworkdb.py
@@ -26,13 +26,18 @@
 import logging
 import fnmatch
 import os
+import sys
 import re
 
 import gv
 import yaml
 
-import kapidox.depdiagram.gvutils
-from kapidox.depdiagram.framework import Framework
+if sys.version_info.major < 3:
+    import gvutils
+    from framework import Framework
+else:
+    import kapidox.depdiagram.gvutils
+    from kapidox.depdiagram.framework import Framework
 
 
 TARGET_SHAPES = [