diff --git a/addon/classes/draw.py b/addon/classes/draw.py
index a95f53e..ad3fa36 100644
--- a/addon/classes/draw.py
+++ b/addon/classes/draw.py
@@ -1,4 +1,5 @@
 from .. import __package__ as base_package
+import bpy
 
 def draw_addon_separator(layout):
     layout.separator()