Update search to fix mocks