Fix game logic