Update score after undos

1 files changed, 3 insertions, 1 deletions

diff --git a/gogame.py b/gogame.py
index 01390ba..51a9486 100644
--- a/gogame.py
+++ b/gogame.py
@@ -192,7 +192,9 @@ class GoGame:
                 pos, color, captures = self.undostack.pop()
                 del self.status[pos]
                 if captures:
-                    for p in captures: self.status[p] = self.invert(color)
+                    for p in captures: 
+                        self.status[p] = self.invert(color)
+                        self.score[color] -= 1
                 return True
             else:
                 return False