From c0eb9addfdb514780f96dd78302b2b431e8a4da4 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Fri, 19 Nov 2021 19:13:21 +1100
Subject: [PATCH] Fix bug in FM utility for grabbing classes from the draft
 FMHEM standard

---
 src/ifcopenshell-python/ifcopenshell/util/fm.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/ifcopenshell-python/ifcopenshell/util/fm.py b/src/ifcopenshell-python/ifcopenshell/util/fm.py
index cad7482fab..27788ce309 100644
--- a/src/ifcopenshell-python/ifcopenshell/util/fm.py
+++ b/src/ifcopenshell-python/ifcopenshell/util/fm.py
@@ -114,6 +114,10 @@ def get_cobie_components(ifc_file):
 
 def get_fmhem_types(ifc_file):
     elements = []
+    if ifc_file.schema == "IFC2X3":
+        fmhem_classes = fmhem_classes_ifc2x3
+    else:
+        fmhem_classes = fmhem_classes_ifc4
     for ifc_class in fmhem_classes:
         try:
             if ifc_class == "IfcEnergyConversionDeviceType":