Fix bug with transformers