Bug correction end of game