diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
index ec24c89..31e7fa2 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
@@ -197,7 +197,7 @@
 			
 			// Declare input methods in resources and the main type.
 			for (ChannelGenerator cg : model.getIOChannelGenerators()) {
-				for (ChannelMember cm : cg.getChannelMembers()) {
+				for (ChannelMember cm : ((DataflowChannelGenerator) cg).getOutputChannelMembers()) {
 					if (cm.getIdentifierTemplate().equals(rn.getIdentifierTemplate())) {
 						Expression message = cm.getStateTransition().getMessageExpression();
 						if (message.getClass() == Term.class) {