Update search_test.go