Fix missing import word_counts_in_set.