diff --git a/board.py b/board.py
index 092b10c..10d071e 100644
--- a/board.py
+++ b/board.py
@@ -21,7 +21,7 @@ class Board:
         return self.state
 
     def switch(self,cur):
-        return -1 if cur == 1 else 1
+        return -1 * cur
 
 
     # Remember to handle edge case when we're on the last moves and you may go