diff --git a/src/com/interview/suffixprefix/Trie.java b/src/com/interview/suffixprefix/Trie.java
index d91dc8f1..5077161c 100644
--- a/src/com/interview/suffixprefix/Trie.java
+++ b/src/com/interview/suffixprefix/Trie.java
@@ -140,7 +140,7 @@ private boolean delete(TrieNode current, String word, int index) {
         if (shouldDeleteCurrentNode) {
             current.children.remove(ch);
             //return true if no mappings are left in the map.
-            return current.children.size() == 0;
+            return (current.children.size() == 0 && !current.endOfWord);
         }
         return false;
     }