End Screen fix