diff --git a/go_player/myPlayer.py b/go_player/myPlayer.py
index 63139d2..72a0786 100644
--- a/go_player/myPlayer.py
+++ b/go_player/myPlayer.py
@@ -198,7 +198,10 @@ class myPlayer(PlayerInterface):
         with torch.no_grad():
             prediction = self.model(go_board)
 
-        return prediction
+        if color == Goban.Board._BLACK:
+            return prediction
+
+        return 1 - prediction
 
     def getPlayerMove(self):
         if self._board.is_game_over():