Minor UX fix