diff --git a/src/bonsai/bonsai/tool/type.py b/src/bonsai/bonsai/tool/type.py
index 6c98e8bb59..43d86d5076 100644
--- a/src/bonsai/bonsai/tool/type.py
+++ b/src/bonsai/bonsai/tool/type.py
@@ -74,11 +74,13 @@ class Type(bonsai.core.tool.Type):
     def get_model_types(cls) -> list[ifcopenshell.entity_instance]:
         ifc_file = tool.Ifc.get()
         types = ifc_file.by_type("IfcElementType")
-        types += ifc_file.by_type("IfcSpatialElementType")
-        types += ifc_file.by_type("IfcTypeProduct", include_subtypes=False)
-        if not tool.Ifc.get_schema().startswith("IFC4X3"):
+        if tool.Ifc.get_schema() == "IFC2X3":
             types += ifc_file.by_type("IfcWindowStyle")
             types += ifc_file.by_type("IfcDoorStyle")
+            types += ifc_file.by_type("IfcSpatialStructureElementType")
+        else:
+            types += ifc_file.by_type("IfcSpatialElementType")
+        types += ifc_file.by_type("IfcTypeProduct", include_subtypes=False)
         return types
 
     @classmethod