Fix end_screen module