Small bug fix for search