diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
index de89ba5..4941a19 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
@@ -218,6 +218,17 @@
 									}
 								}
 							}
+						} else if (message.getClass() == Variable.class) {
+							MethodDeclaration input = new MethodDeclaration(
+									((Variable) cm.getStateTransition().getMessageExpression()).getName(),
+									false, typeVoid, null);
+							type.addMethod(input);
+							String str = ((Variable) cm.getStateTransition().getMessageExpression()).getName();
+							input = getMethod(mainType, str);
+							if (input == null) {
+								input = new MethodDeclaration(str, false, typeVoid, null);
+								mainType.addMethod(input);
+							}
 						}
 					}
 				}