diff --git a/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java b/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java
index 65d20f0..4083dce 100644
--- a/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java
+++ b/AlgebraicDataflowArchitectureModel/src/application/actions/JavaPrototypeGenerateAction.java
@@ -18,6 +18,7 @@
 import generators.JavaCodeGenerator;
 import generators.JavaMethodBodyGenerator;
 import generators.JavaSpecific;
+import models.dataConstraintModel.ResourceHierarchy;
 import models.dataConstraintModel.ResourcePath;
 import models.dataFlowModel.DataTransferModel;
 import models.dataFlowModel.ModelExtension;
@@ -47,8 +48,8 @@
 			if (fileName == null) fileName = "Main";
 			String mainTypeName = fileName.split("\\.")[0];
 			boolean exist = false;
-			for (ResourcePath id: model.getResourcePaths()) {
-				String resourceName = id.getResourceName().substring(0, 1).toUpperCase() + id.getResourceName().substring(1);
+			for (ResourceHierarchy res: model.getResourceHierarchies()) {
+				String resourceName = res.getResourceName().substring(0, 1).toUpperCase() + res.getResourceName().substring(1);
 				if (mainTypeName.equals(resourceName)) {
 					exist = true;
 				}