Update the code for recent search_api changes.