Reset after game over