Minor UI enhancement