Update for transformers-0.4