Minor fix ui