End game works again!