diff --git a/design-patterns/src/main/java/org/prateek/BehaviouralPatterns/MementoPattern/Caretaker.java b/design-patterns/src/main/java/org/prateek/BehaviouralPatterns/MementoPattern/Caretaker.java
index 56f7f8b..ab63dc8 100644
--- a/design-patterns/src/main/java/org/prateek/BehaviouralPatterns/MementoPattern/Caretaker.java
+++ b/design-patterns/src/main/java/org/prateek/BehaviouralPatterns/MementoPattern/Caretaker.java
@@ -10,7 +10,7 @@ public void saveState(TextEditor editor){
         history.push(editor.save());
     }
     public void undo(TextEditor editor){
-        if(!history.empty()){
+        if(history.size()>1){
           history.pop();
           editor.restore(history.peek());
         }