diff --git a/README.md b/README.md
index d74f88c..cbdfbc0 100644
--- a/README.md
+++ b/README.md
@@ -35,7 +35,7 @@ sentiment_task("T'estimo!")
 ```python
 from transformers import AutoModelForSequenceClassification
 from transformers import TFAutoModelForSequenceClassification
-from transformers import AutoTokenizer
+from transformers import AutoTokenizer, AutoConfig
 import numpy as np
 from scipy.special import softmax