[search] Add missing import