Fix search box implementation