Tiny fix for the search (step 3)