Fix search ui tests