Fix regression in search

Test Plan:
- Do a search, make sure it works