diff --git a/src/Mod/Draft/drafttests/test_dwg.py b/src/Mod/Draft/drafttests/test_dwg.py
index 3a62989023..195d6f9135 100644
--- a/src/Mod/Draft/drafttests/test_dwg.py
+++ b/src/Mod/Draft/drafttests/test_dwg.py
@@ -36,6 +36,7 @@ import Draft
 import drafttests.auxiliary as aux
 
 from draftutils.messages import _msg
+from drafttests import test_base
 
 
 class DraftDWG(test_base.DraftTestCaseDoc):