Fixed bug in game logic

All's good