little fix in ui