diff --git a/de.prob.units.tests/src/de/prob/units/tests/PragmaTranslatorTest.java b/de.prob.units.tests/src/de/prob/units/tests/PragmaTranslatorTest.java
index f2b199e665188d58e992068aa05f44c0012b61c1..afb9b77b83c1f68964af1e69f01f319492208983 100644
--- a/de.prob.units.tests/src/de/prob/units/tests/PragmaTranslatorTest.java
+++ b/de.prob.units.tests/src/de/prob/units/tests/PragmaTranslatorTest.java
@@ -87,7 +87,7 @@ public class PragmaTranslatorTest extends AbstractEventBTests {
 		TranslatorFactory.translate(context, writer);
 
 		assertEquals(
-				"package(load_event_b_project([],[event_b_context(none,'TestContext',[extends(none,[]),constants(none,[identifier(none,cst1)]),axioms(none,[equal(rodinpos('TestContext',axm1,'('),identifier(none,cst1),integer(none,5))]),theorems(none,[]),sets(none,[])])],[exporter_version(2)],_Error)).\n",
+				"package(load_event_b_project([],[event_b_context(none,'TestContext',[extends(none,[]),constants(none,[identifier(none,cst1)]),axioms(none,[equal(rodinpos('TestContext',axm1,'('),identifier(none,cst1),integer(none,5))]),theorems(none,[]),sets(none,[])])],[exporter_version(2),pragma(unit,'TestContext',cst1,[test])],_Error)).\n",
 				stringWriter.getBuffer().toString());
 	}
 }