Small bug fix in game state