Fix search display issue