End screen fix