Fixed UI bug in search