diff --git a/course/fr/chapter3/section4.ipynb b/course/fr/chapter3/section4.ipynb
index eaa09799..6538d07e 100644
--- a/course/fr/chapter3/section4.ipynb
+++ b/course/fr/chapter3/section4.ipynb
@@ -64,7 +64,7 @@
    },
    "outputs": [],
    "source": [
-    "tokenized_datasets = tokenized_datasets.remove_columns([\"sentence1\", \"sentence2\", \"idx\"])\n",
+    "tokenized_datasets = tokenized_datasets.remove_columns([\"sentence1\", \"sentence2\", \"id\"])\n",
     "tokenized_datasets = tokenized_datasets.rename_column(\"label\", \"labels\")\n",
     "tokenized_datasets.set_format(\"torch\")\n",
     "tokenized_datasets[\"train\"].column_names"
@@ -145,7 +145,7 @@
    },
    "outputs": [],
    "source": [
-    "from transformers import AdamW\n",
+    "from torch.optim import AdamW\n",
     "\n",
     "optimizer = AdamW(model.parameters(), lr=5e-5)"
    ]
@@ -220,9 +220,10 @@
    },
    "outputs": [],
    "source": [
-    "from datasets import load_metric\n",
+    "!pip install evaluate\n",
+    "import evaluate",
     "\n",
-    "metric = load_metric(\"glue\", \"mrpc\")\n",
+    "metric = evaluate.load(\"glue\", \"mrpc\")\n",
     "model.eval()\n",
     "for batch in eval_dataloader:\n",
     "    batch = {k: v.to(device) for k, v in batch.items()}\n",