Quick fix for search functionality