1 files changed, 3 insertions, 4 deletions
diff --git a/src/bot.cpp b/src/bot.cpp
index 0e959d0..7abe516 100644
--- a/src/bot.cpp
+++ b/src/bot.cpp
@@ -154,10 +154,9 @@ static int search(int depth, Game *b, int alpha, int beta, int ply) {
   }
   const uint64_t gameHash = GenerateZobristKey(b);
 
-  auto repIt = b->ThreeFoldMap.find(gameHash);
-  if (repIt != b->ThreeFoldMap.end() && repIt->second >= 2) {
-    return 0; // repetition -> draw
-  }
+  if (isRepetionDraw(gameHash, b)) {
+    return 0;
+  };
 
   TranspositionsEntry *entry = nullptr;
   if (auto it = b->Transpositions->find(gameHash);