Minor fix on UI.