diff --git a/.gitignore b/.gitignore
index 64e4208..85c7f25 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 /stockfish/
+.idea
\ No newline at end of file
diff --git a/eval.py b/eval.py
index 99076d0..cda3c56 100644
--- a/eval.py
+++ b/eval.py
@@ -116,7 +116,7 @@ def check_endgame(board: chess.Board) -> bool:
     minors_white = 0
     queens_black = 0
     minors_black = 0
-    for s in chess.Square:
+    for s in chess.SQUARES:
         piece = board.piece_at(s)
         if piece is None:
             continue