X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fgamestate.py;fp=naja%2Fgamestate.py;h=96e55056bf862b5243aed0198bd46e8a0ce097a7;hb=62a297a628fe85160afa770ecad666e3d1a50126;hp=9dca03026f773869e126e71647358dbfdba354d1;hpb=78a001f6b445795f347d1ae7399d8f3b4072f779;p=naja.git

diff --git a/naja/gamestate.py b/naja/gamestate.py
index 9dca030..96e5505 100644
--- a/naja/gamestate.py
+++ b/naja/gamestate.py
@@ -2,7 +2,7 @@
 The current game state.
 """
 
-from .constants import NORTH, SOUTH, EAST, WEST
+from .constants import BITS
 from .gameboard import GameBoard
 from .player import Player
 
@@ -12,7 +12,10 @@ class GameState(object):
     Naja game state.
     """
 
-    INITIAL_BITS = NORTH | SOUTH | EAST | WEST
+    INITIAL_BITS = (
+        BITS.NORTH | BITS.SOUTH |
+        BITS.EAST | BITS.WEST
+    )
     MAX_HEALTH = 4
     WINS_REQUIRED = 4