diff --git a/de.prob.units/plugin.xml b/de.prob.units/plugin.xml
index 4656bfef75443c19efa4961d7bf3fd4d2196ec2c..7c8fbc3eb4ae41cd56a219597d022f5e62ec8dd7 100644
--- a/de.prob.units/plugin.xml
+++ b/de.prob.units/plugin.xml
@@ -39,6 +39,24 @@
             style="single"
             typeId="de.prob.units.inferredUnitPragmaAttribute">
       </textAttribute>
+      <attributeRelation
+            elementTypeId="org.eventb.core.scVariable">
+         <attributeReference
+               descriptionId="de.prob.units.unitPragmaAttribute">
+         </attributeReference>
+         <attributeReference
+               descriptionId="de.prob.units.inferredUnitPragmaAttribute">
+         </attributeReference>
+      </attributeRelation>
+      <attributeRelation
+            elementTypeId="org.eventb.core.scConstant">
+         <attributeReference
+               descriptionId="de.prob.units.unitPragmaAttribute">
+         </attributeReference>
+         <attributeReference
+               descriptionId="de.prob.units.inferredUnitPragmaAttribute">
+         </attributeReference>
+      </attributeRelation>
    </extension>
    <extension
          point="org.eclipse.ui.handlers">