Fix the search box behavior