Fix search UI