Change vocab_entailment back to vocab_expansion.