diff --git a/src/Mod/Arch/ArchComponent.py b/src/Mod/Arch/ArchComponent.py
index 8cf765f99a..a8128378c8 100644
--- a/src/Mod/Arch/ArchComponent.py
+++ b/src/Mod/Arch/ArchComponent.py
@@ -1043,7 +1043,7 @@ class ComponentTaskPanel:
         try:
             import BimClassification
         except:
-            self.form.ButtonCode.hide()
+            self.classButton.hide()
         else:
             import os
             self.classButton.setIcon(QtGui.QIcon(os.path.join(os.path.dirname(BimClassification.__file__),"icons","BIM_Classification.svg")))