diff --git a/de.prob.core/plugin.xml b/de.prob.core/plugin.xml
index b84a1eaef2043d86514040b1a698b95d501b759d..eaeb084496924b1941929b0cf893552efd41915e 100644
--- a/de.prob.core/plugin.xml
+++ b/de.prob.core/plugin.xml
@@ -23,4 +23,12 @@
             class="de.prob.core.StaticListenerRegistry">
       </listener>
    </extension>
+   <extension
+         point="org.rodinp.core.attributeTypes">
+      <attributeType
+            id="unitPragma"
+            kind="string"
+            name="Content of a unit Pragma to send to ProB">
+      </attributeType>
+   </extension>
 </plugin>
diff --git a/de.prob.ui/plugin.xml b/de.prob.ui/plugin.xml
index a64882ef7a42135a4a34f5d12d6b38230c1960c2..696b98996d9a672f5c02f29818fe2778c05910d8 100644
--- a/de.prob.ui/plugin.xml
+++ b/de.prob.ui/plugin.xml
@@ -1195,4 +1195,21 @@
          </variable>
       </sourceProvider>
         </extension>
+   <extension
+         point="org.eventb.ui.editorItems">
+      <textAttribute
+            class="de.prob.ui.pragmas.UnitPragma"
+            expandsHorizontally="true"
+            id="de.prob.ui.unitPragma"
+            isMath="true"
+            style="de.prob.ui.unitPragma"
+            typeId="de.prob.core.unitPragma">
+      </textAttribute>
+      <attributeRelation
+            elementTypeId="org.eventb.core.event">
+         <attributeReference
+               descriptionId="de.prob.ui.unitPragma">
+         </attributeReference>
+      </attributeRelation>
+   </extension>
 </plugin>