Minor fix for UI