From b4d633a34d5574ade69af9b8cc6f39a0ad2a7b33 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Fri, 3 Feb 2023 21:28:36 +1100
Subject: [PATCH] Fix #2735.

---
 src/blenderbim/blenderbim/bim/module/drawing/prop.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/blenderbim/blenderbim/bim/module/drawing/prop.py b/src/blenderbim/blenderbim/bim/module/drawing/prop.py
index 6d2ee22d66..51ac3689a1 100644
--- a/src/blenderbim/blenderbim/bim/module/drawing/prop.py
+++ b/src/blenderbim/blenderbim/bim/module/drawing/prop.py
@@ -75,8 +75,14 @@ def update_diagram_scale(self, context):
     if "|" not in scale:
         return
     human_scale, scale = scale.split("|")
-    element = tool.Ifc.get_entity(self.id_data)
-    if not element:
+    try:
+        element = (
+            tool.Ifc.get()
+            .by_id(self.id_data.BIMMeshProperties.ifc_definition_id)
+            .OfProductRepresentation[0]
+            .ShapeOfProduct[0]
+        )
+    except:
         return
     pset = ifcopenshell.util.element.get_psets(element).get("EPset_Drawing")
     if pset: