diff --git a/src/bonsai/bonsai/tool/drawing.py b/src/bonsai/bonsai/tool/drawing.py
index 096ec7b275..a021b608c4 100644
--- a/src/bonsai/bonsai/tool/drawing.py
+++ b/src/bonsai/bonsai/tool/drawing.py
@@ -1850,7 +1850,7 @@ class Drawing(bonsai.core.tool.Drawing):
         if include:
             elements = ifcopenshell.util.selector.filter_elements(ifc_file, include)
         else:
-            if tool.Ifc.get_schema() == "IFC2X3":
+            if ifc_file.schema == "IFC2X3":
                 base_elements = set(ifc_file.by_type("IfcElement") + ifc_file.by_type("IfcSpatialStructureElement"))
             else:
                 base_elements = set(ifc_file.by_type("IfcElement") + ifc_file.by_type("IfcSpatialElement"))