Fixed problem with search