Fix game not restarting properly