diff --git a/src/bonsai/bonsai/bim/module/root/data.py b/src/bonsai/bonsai/bim/module/root/data.py
index 64edaa2a9d..99b17220b5 100644
--- a/src/bonsai/bonsai/bim/module/root/data.py
+++ b/src/bonsai/bonsai/bim/module/root/data.py
@@ -97,7 +97,7 @@ class IfcClassData:
         return types_enum
 
     @classmethod
-    def ifc_classes_suggestions(cls):
+    def ifc_classes_suggestions(cls) -> dict[str, list[dict[str, Union[str, None]]]]:
         # suggestions : dict[class_name: list[dict[predefined_type, name(optional)]]]
         suggestions = defaultdict(list)
         version = tool.Ifc.get_schema()
diff --git a/src/bonsai/bonsai/bim/module/root/prop.py b/src/bonsai/bonsai/bim/module/root/prop.py
index 4c358dfde1..e2d6cea330 100644
--- a/src/bonsai/bonsai/bim/module/root/prop.py
+++ b/src/bonsai/bonsai/bim/module/root/prop.py
@@ -99,7 +99,7 @@ def get_ifc_classes(self: "BIMRootProperties", context: bpy.types.Context) -> li
     return IfcClassData.data["ifc_classes"]
 
 
-def get_ifc_classes_suggestions(self: "BIMRootProperties", context: bpy.types.Context) -> list[tuple[str, str]]:
+def get_ifc_classes_suggestions() -> dict[str, list[dict[str, Union[str, None]]]]:
     if not IfcClassData.is_loaded:
         IfcClassData.load()
     return IfcClassData.data["ifc_classes_suggestions"]