diff --git a/AlgebraicDataflowArchitectureModel/src/algorithm/DataStorageDecision.java b/AlgebraicDataflowArchitectureModel/src/algorithm/DataStorageDecision.java
index b6576d6..78df79b 100644
--- a/AlgebraicDataflowArchitectureModel/src/algorithm/DataStorageDecision.java
+++ b/AlgebraicDataflowArchitectureModel/src/algorithm/DataStorageDecision.java
@@ -16,7 +16,7 @@
 	}
 
 	static private void trackNode(ResourceNode resource) {
-		if(arrivedNode.contains(resource)) return;
+		if (arrivedNode.contains(resource)) return;
 		arrivedNode.add(resource);
 		boolean flag = false;
 		for (Edge e : resource.getInEdges()) {
@@ -25,6 +25,7 @@
 				flag = true;
 			}
 		}
+		if (resource.getInEdges().size() == 0) flag = true;
 		((StoreAttribute) resource.getAttribute()).setStored(flag);
 	}
 }