Fix game restart