fix bug UI