Minor UI tweek