diff --git a/src/main/java/BattleShip/GridGUI.java b/src/main/java/BattleShip/GridGUI.java
index d94c26a..158f734 100644
--- a/src/main/java/BattleShip/GridGUI.java
+++ b/src/main/java/BattleShip/GridGUI.java
@@ -109,7 +109,15 @@ public class GridGUI {
                         gameOver = false;
                     }
                 }
+                if(gameOver) {
+                    text = "You win! You took " + numOfGuesses + " guesses.";
+                    endGame = true;
+                }
+
+                clicked = true;
             }
         }
     }
-}
\ No newline at end of file
+}
+
+