diff --git a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
index 368aa36..00cc669 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
@@ -266,6 +266,18 @@
 				termEntItr = subTerms.entrySet().iterator();
 			}
 		}
+		// Replace the type of the state field.
+		Type fieldType = JavaCodeGenerator.getImplStateType(childRes.getParent());
+		if (exp instanceof Term) {
+			((Term) exp).setType(fieldType);
+			for (Map.Entry<Position, Variable> varEnt: ((Term) exp).getVariables().entrySet()) {
+				if (varEnt.getValue().getName().equals(fieldOfResourceState)) {
+					varEnt.getValue().setType(fieldType);
+				}
+			}
+		} else if (exp instanceof Variable) {
+			((Variable) exp).setType(fieldType);
+		}
 		String[] sideEffects = new String[] {""};
 		String newState = exp.toImplementation(sideEffects);
 		String updateStatement;
diff --git a/AlgebraicDataflowArchitectureModel/src/generators/JavaMethodBodyGenerator.java b/AlgebraicDataflowArchitectureModel/src/generators/JavaMethodBodyGenerator.java
index fa8e030..cbe262c 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/JavaMethodBodyGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/JavaMethodBodyGenerator.java
@@ -121,6 +121,18 @@
 												replaceJsonTermWithConstructorInvocation(updateExp, childStateType, childComponentName, childComponent);
 											}
 										}
+										// Replace the type of the state field.
+										Type fieldType = JavaCodeGenerator.getImplStateType(outRes);
+										if (updateExp instanceof Term) {
+											((Term) updateExp).setType(fieldType);
+											for (Map.Entry<Position, Variable> varEnt: ((Term) updateExp).getVariables().entrySet()) {
+												if (varEnt.getValue().getName().equals("value")) {
+													varEnt.getValue().setType(fieldType);
+												}
+											}
+										} else if (updateExp instanceof Variable) {
+											((Variable) updateExp).setType(fieldType);
+										}
 										// Add statements to the update method.
 										String[] sideEffects = new String[] {""};
 										String newState = updateExp.toImplementation(sideEffects);
@@ -691,6 +703,18 @@
 									replaceJsonTermWithConstructorInvocation(updateExp, childStateType, childComponentName, childComponent);
 								}
 							}
+							// Replace the type of the state field.
+							Type fieldType = JavaCodeGenerator.getImplStateType(outRes);
+							if (updateExp instanceof Term) {
+								((Term) updateExp).setType(fieldType);
+								for (Map.Entry<Position, Variable> varEnt: ((Term) updateExp).getVariables().entrySet()) {
+									if (varEnt.getValue().getName().equals("value")) {
+										varEnt.getValue().setType(fieldType);
+									}
+								}
+							} else if (updateExp instanceof Variable) {
+								((Variable) updateExp).setType(fieldType);
+							}
 							// Add statements to the input method.
 							String[] sideEffects = new String[] {""};
 							String newState = updateExp.toImplementation(sideEffects);
diff --git a/AlgebraicDataflowArchitectureModel/src/generators/JerseyMethodBodyGenerator.java b/AlgebraicDataflowArchitectureModel/src/generators/JerseyMethodBodyGenerator.java
index 9eebb7f..a77cca1 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/JerseyMethodBodyGenerator.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/JerseyMethodBodyGenerator.java
@@ -141,6 +141,18 @@
 												replaceJsonTermWithConstructorInvocation(updateExp, childStateType, childComponentName, childComponent);
 											}
 										}
+										// Replace the type of the state field.
+										Type fieldType = JerseyCodeGenerator.getImplStateType(outRes);
+										if (updateExp instanceof Term) {
+											((Term) updateExp).setType(fieldType);
+											for (Map.Entry<Position, Variable> varEnt: ((Term) updateExp).getVariables().entrySet()) {
+												if (varEnt.getValue().getName().equals("value")) {
+													varEnt.getValue().setType(fieldType);
+												}
+											}
+										} else if (updateExp instanceof Variable) {
+											((Variable) updateExp).setType(fieldType);
+										}
 										// Add statements to the update method.
 										String[] sideEffects = new String[] {""};
 										String newState = updateExp.toImplementation(sideEffects);
@@ -847,6 +859,18 @@
 									replaceJsonTermWithConstructorInvocation(updateExp, childStateType, childComponentName, childComponent);
 								}
 							}
+							// Replace the type of the state field.
+							Type fieldType = JerseyCodeGenerator.getImplStateType(outRes);
+							if (updateExp instanceof Term) {
+								((Term) updateExp).setType(fieldType);
+								for (Map.Entry<Position, Variable> varEnt: ((Term) updateExp).getVariables().entrySet()) {
+									if (varEnt.getValue().getName().equals("value")) {
+										varEnt.getValue().setType(fieldType);
+									}
+								}
+							} else if (updateExp instanceof Variable) {
+								((Variable) updateExp).setType(fieldType);
+							}
 							// Add statements to the input method.
 							String[] sideEffects = new String[] {""};
 							String newState = updateExp.toImplementation(sideEffects);