Minor UI change

Minor UI change