diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
index d75e568..7e4df1b 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/JavaCodeGenerator.java
@@ -198,7 +198,7 @@
 			// Declare input methods in resources and the main type.
 			for (ChannelGenerator cg : model.getIOChannelGenerators()) {
 				for (ChannelMember cm : cg.getChannelMembers()) {
-					if (cm.getIdentifierTemplate().getResourceName().equals(type.getTypeName().toLowerCase())) {
+					if (cm.getIdentifierTemplate().equals(rn.getIdentifierTemplate())) {
 						Expression message = cm.getStateTransition().getMessageExpression();
 						if (message.getClass() == Term.class) {
 							ArrayList<VariableDeclaration> params = new ArrayList<>();