Minor fix

fix on ui.h