Minor UX improvement