diff --git a/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java b/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java
index 4083dce..846c273 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java
@@ -18,6 +18,7 @@
 import generators.JavaCodeGenerator;
 import generators.JavaMethodBodyGenerator;
 import generators.JavaSpecific;
+import generators.StandaloneSpecific;
 import models.dataConstraintModel.ResourceHierarchy;
 import models.dataConstraintModel.ResourcePath;
 import models.dataFlowModel.DataTransferModel;
@@ -59,8 +60,8 @@
 			} else {
 				JavaCodeGenerator.resetMainTypeName();			// use the default main type's name.
 			}
-			editor.setCodes(JavaMethodBodyGenerator.doGenerate(graph, model, JavaCodeGenerator.doGenerate(graph, model)));
-//			editor.setCodes(new CodeGeneratorFromDataFlowGraph().generateCode(model, graph, new JavaSpecific()));
+//			editor.setCodes(JavaMethodBodyGenerator.doGenerate(graph, model, JavaCodeGenerator.doGenerate(graph, model)));
+			editor.setCodes(new CodeGeneratorFromDataFlowGraph().generateCode(model, graph, new StandaloneSpecific(), new JavaSpecific()));
 			ModelExtension.recoverModel(model);
 			for (CompilationUnit file : editor.getCodes()) {
 				System.out.println(file);