Fix generator test, remove search_gen from coverage