Fix bug with the search