diff --git a/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java b/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
index 6f4ff61..69efb90 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithms/CodeGenerator.java
@@ -56,10 +56,8 @@
 			if (((StoreAttribute) rn.getAttribute()).isStored()) {
 				type.addField(new FieldDeclaration(DataConstraintModel.typeInt,
 						rn.getIdentifierTemplate().getResourceName()));
-				type.addMethod(new MethodDeclaration("set" + type.getTypeName(), new Type("Void", "void")));
 			}
-			type.addMethod(new MethodDeclaration("get" + type.getTypeName(),
-					new Type(type.getTypeName(), type.getTypeName())));
+			type.addMethod(new MethodDeclaration("get" + type.getTypeName(),DataConstraintModel.typeInt));
 			codes.add(type);
 		}
 		return codes;