Game-over return function fixed