Another small UI tweek