Small fix with key presses at the end of the game.