tests: fix Search calls in tests