Fixing the game restart bug?

Maybe? Please?