diff --git a/transformers_doc/en/training.ipynb b/transformers_doc/en/training.ipynb
index b5c953210..9dcf8438e 100644
--- a/transformers_doc/en/training.ipynb
+++ b/transformers_doc/en/training.ipynb
@@ -186,7 +186,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from transformers import TrainingArguments\n",
+    "from transformers import TrainingArguments, Trainer\n",
     "\n",
     "training_args = TrainingArguments(\n",
     "    output_dir=\"yelp_review_classifier\",\n",
diff --git a/transformers_doc/es/training.ipynb b/transformers_doc/es/training.ipynb
index 67d5ae710..ac1b8fdd3 100644
--- a/transformers_doc/es/training.ipynb
+++ b/transformers_doc/es/training.ipynb
@@ -303,7 +303,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from transformers import TrainingArguments\n",
+    "from transformers import TrainingArguments, Trainer\n",
     "\n",
     "training_args = TrainingArguments(output_dir=\"test_trainer\", eval_strategy=\"epoch\")"
    ]
diff --git a/transformers_doc/pt/training.ipynb b/transformers_doc/pt/training.ipynb
index f9f119ff9..8b78668e7 100644
--- a/transformers_doc/pt/training.ipynb
+++ b/transformers_doc/pt/training.ipynb
@@ -330,7 +330,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from transformers import TrainingArguments\n",
+    "from transformers import TrainingArguments, Trainer\n",
     "\n",
     "training_args = TrainingArguments(output_dir=\"test_trainer\", eval_strategy=\"epoch\")"
    ]