Fix finish game after selecting play again