diff --git a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
index ccea9b1..fc6b3f4 100644
--- a/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
+++ b/AlgebraicDataflowArchitectureModel/src/generators/CodeGeneratorFromDataFlowGraph.java
@@ -730,12 +730,13 @@
 					children.add(child.getResourceHierarchy());
 					List<VariableDeclaration> params = new ArrayList<>();
 					int v = 1;
-					for (Selector param: child.getSelectors()) {
-						if (param.getExpression() instanceof Variable) {
-							Variable var = (Variable) param.getExpression();
+					Expression param = child.getPrimaryResourcePath().getLastParam();
+					if (param != null) {
+						if (param instanceof Variable) {
+							Variable var = (Variable) param;
 							params.add(langSpec.newVariableDeclaration(var.getType(), var.getName()));
-						} else if (param.getExpression() instanceof Term) {
-							Term var = (Term) param.getExpression();
+						} else if (param instanceof Term) {
+							Term var = (Term) param;
 							params.add(langSpec.newVariableDeclaration(var.getType(), "v" + v));
 						}
 						v++;