Minor fix on U.I