Fix game ending