get_current_screen fix