The custom_search method wasn't working correctly.
Added a test for it to make sure it works.