Fix search ui integration