diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
index e4ab063..3a5da26 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
@@ -123,6 +123,9 @@
 									((Term) cm.getStateTransition().getMessageExpression()).getSymbol().getImplName(),
 									false, typeVoid, params);
 							type.addMethod(io);
+							io = new MethodDeclaration(
+									((Term) cm.getStateTransition().getMessageExpression()).getSymbol().getImplName(),
+									false, typeVoid, params);
 							codes.get(0).addMethod(io);
 						}
 					}