From 021e6b9ef5b9ff60fd733570adde3d0254063f83 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Mon, 16 Mar 2026 15:15:12 +1100
Subject: [PATCH] Simplify get model types to just got all type products. Fixes
 failing test.

---
 src/bonsai/bonsai/tool/type.py | 11 +----------
 1 file changed, 1 insertion(+), 10 deletions(-)

diff --git a/src/bonsai/bonsai/tool/type.py b/src/bonsai/bonsai/tool/type.py
index 84a349f288..882c4b4618 100644
--- a/src/bonsai/bonsai/tool/type.py
+++ b/src/bonsai/bonsai/tool/type.py
@@ -75,16 +75,7 @@ class Type(bonsai.core.tool.Type):
 
     @classmethod
     def get_model_types(cls) -> list[ifcopenshell.entity_instance]:
-        ifc_file = tool.Ifc.get()
-        types = ifc_file.by_type("IfcElementType")
-        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
+        return tool.Ifc.get().by_type("IfcTypeProduct")
 
     @classmethod
     def get_object_data(cls, obj: bpy.types.Object) -> Union[bpy.types.ID, None]: