1 files changed, 3 insertions, 1 deletions
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Stanik.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Stanik.java
index 6c0f380..6761c94 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Stanik.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Stanik.java
@@ -94,6 +94,8 @@ public class Stanik extends Module {
     private boolean pruneZMI(ZMI zmi, ValueTime time) {
         Value timestamp = zmi.getAttributes().get("timestamp");
 
+        boolean isLeaf = zmi.getSons().isEmpty();
+
         List<ZMI> sonsToRemove = new LinkedList();
         if (ValueUtils.valueLower(timestamp, time.subtract(new ValueDuration(freshnessPeriod)))) {
             if (zmi.getFather() != null) {
@@ -111,7 +113,7 @@ public class Stanik extends Module {
             zmi.removeSon(son);
         }
 
-        if (zmi.getSons().isEmpty()) {
+        if (!isLeaf && zmi.getSons().isEmpty()) {
             return true;
         }