Fix game restart
update