diff --git a/src/blenderbim/blenderbim/bim/module/drawing/prop.py b/src/blenderbim/blenderbim/bim/module/drawing/prop.py
index a0935d6ac7..c56e6b9262 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.scene.active_object)
+    element = tool.Ifc.get_entity(context.active_object)
     if not element:
         return
     pset = ifcopenshell.util.element.get_psets(element).get("EPset_Drawing")