[UPDATES] Fixing end dialog