diff --git a/src/main/java/com/shr4pnel/minesweeper/Controller.java b/src/main/java/com/shr4pnel/minesweeper/Controller.java
index 3b405ff..d63ec49 100644
--- a/src/main/java/com/shr4pnel/minesweeper/Controller.java
+++ b/src/main/java/com/shr4pnel/minesweeper/Controller.java
@@ -293,7 +293,7 @@ public class Controller {
                 Button current = (Button) getNodeByRowColumnIndex(row, column);
                 assert current != null;
                 String currentURL = getButtonURL(current);
-                if (currentURL.contains("blank.png") || currentURL.contains("flag.png")) {
+                if (currentURL.contains("blank.png") || currentURL.contains("flagged.png")) {
                     unrevealedTiles++;
                 }
             }