Fix win/lose game logic