Small UX change