Fix search tests
Fix search tests