diff --git a/bundles/org.eclipse.vorto.codegen.ui/src/org/eclipse/vorto/codegen/ui/wizard/generation/templates/GeneratorTemplate.xtend b/bundles/org.eclipse.vorto.codegen.ui/src/org/eclipse/vorto/codegen/ui/wizard/generation/templates/GeneratorTemplate.xtend
index 1c31fdd781..519370ed96 100644
--- a/bundles/org.eclipse.vorto.codegen.ui/src/org/eclipse/vorto/codegen/ui/wizard/generation/templates/GeneratorTemplate.xtend
+++ b/bundles/org.eclipse.vorto.codegen.ui/src/org/eclipse/vorto/codegen/ui/wizard/generation/templates/GeneratorTemplate.xtend
@@ -38,17 +38,17 @@ class GeneratorTemplate implements IFileTemplate<IGeneratorProjectContext> {
 			import org.eclipse.vorto.codegen.api.GenerationResultZip
 			import org.eclipse.vorto.codegen.api.GeneratorTaskFromFileTemplate
 			import org.eclipse.vorto.codegen.api.IFileTemplate
-			import org.eclipse.vorto.codegen.api.IMappingContext
+			import org.eclipse.vorto.codegen.api.mapping.InvocationContext;
 			import org.eclipse.vorto.codegen.api.IVortoCodeGenerator
 			import org.eclipse.vorto.core.api.model.informationmodel.InformationModel
 			
 			class «context.generatorName» implements IVortoCodeGenerator {
 			
-				override generate(InformationModel infomodel, IMappingContext mappingContext) {
+				override generate(InformationModel infomodel, InvocationContext context) {
 					var output = new GenerationResultZip(infomodel,getServiceKey());
 					var generator = new ChainedCodeGeneratorTask<InformationModel>();
 					generator.addTask(new GeneratorTaskFromFileTemplate(new SampleTemplate()));
-					generator.generate(infomodel,mappingContext,output);
+					generator.generate(infomodel,context,output);
 					return output
 				}
 				
@@ -61,7 +61,7 @@ class GeneratorTemplate implements IFileTemplate<IGeneratorProjectContext> {
 						return "output"
 					}
 						
-					override getContent(InformationModel context) {
+					override getContent(InformationModel model, InvocationContext context) {
 				   		return '//Generated by «context.generatorName»';
 				    }
 				}