Keep compatibility with transformers 0.2.1.