fixed bug with end screen