Fixes search functionality