Fix some more end game logic