diff --git a/de.fhmuenster.masterthesis.textgeneratordsl/src/de/fhmuenster/masterthesis/TestgeneratorDSL.xtext b/de.fhmuenster.masterthesis.textgeneratordsl/src/de/fhmuenster/masterthesis/TestgeneratorDSL.xtext
index d4ebf77c1e8c11fc34ad89b12d0d43e519b8e7fa..d18bc7dbb141224d472fe28ff2c7bca55a33502f 100644
--- a/de.fhmuenster.masterthesis.textgeneratordsl/src/de/fhmuenster/masterthesis/TestgeneratorDSL.xtext
+++ b/de.fhmuenster.masterthesis.textgeneratordsl/src/de/fhmuenster/masterthesis/TestgeneratorDSL.xtext
@@ -152,6 +152,7 @@ FlowReference:
 Test:
 	'Test' name=ID 'for' flowReference=FlowReference 
 		('with' 'priority' priority=INT)?
+		('with' 'flag' ASSOCIATION flag=Flag)?
 		(declarations+=VariableDeclarations (SEPARATOR declarations+=VariableDeclarations)*)?
 		(endCheck=EndChecks)?
 		('with' 'mocks' ASSOCIATION LIST_OPEN (mocks+=MockReference (SEPARATOR mocks+=MockReference)*)? LIST_CLOSED)?