fix showing end screen