Minor fix to the UI.