Update search unit tests