diff --git a/src/bonsai/bonsai/bim/module/model/prop.py b/src/bonsai/bonsai/bim/module/model/prop.py
index f288b6b6cf..31a0f9814c 100644
--- a/src/bonsai/bonsai/bim/module/model/prop.py
+++ b/src/bonsai/bonsai/bim/module/model/prop.py
@@ -18,6 +18,7 @@
 
 import bpy
 import ifcopenshell
+import ifcopenshell.util.element
 import bonsai.tool as tool
 import math
 from bonsai.bim.prop import ObjProperty
diff --git a/src/bonsai/bonsai/bim/module/model/workspace.py b/src/bonsai/bonsai/bim/module/model/workspace.py
index 02c8783959..5ac9aaf45c 100644
--- a/src/bonsai/bonsai/bim/module/model/workspace.py
+++ b/src/bonsai/bonsai/bim/module/model/workspace.py
@@ -316,6 +316,8 @@ class BIM_MT_add_representation_item(Menu):
 
 
 class CreateObjectUI:
+    layout: bpy.types.UILayout
+
     @classmethod
     def draw(cls, context, layout, ifc_element_type=None):
         cls.layout = layout