diff --git a/src/blenderbim/blenderbim/bim/module/drawing/prop.py b/src/blenderbim/blenderbim/bim/module/drawing/prop.py
index b20f871d11..6d2ee22d66 100644
--- a/src/blenderbim/blenderbim/bim/module/drawing/prop.py
+++ b/src/blenderbim/blenderbim/bim/module/drawing/prop.py
@@ -75,7 +75,7 @@ def update_diagram_scale(self, context):
     if "|" not in scale:
         return
     human_scale, scale = scale.split("|")
-    element = tool.Ifc.get_entity(context.active_object)
+    element = tool.Ifc.get_entity(self.id_data)
     if not element:
         return
     pset = ifcopenshell.util.element.get_psets(element).get("EPset_Drawing")