Bugfix for restarting game