diff --git a/course/fr/chapter2/section3_pt.ipynb b/course/fr/chapter2/section3_pt.ipynb
index f14bd85d..9232562c 100644
--- a/course/fr/chapter2/section3_pt.ipynb
+++ b/course/fr/chapter2/section3_pt.ipynb
@@ -111,7 +111,7 @@
    "source": [
     "import torch\n",
     "\n",
-    "model_inputs = torch.tensor(encoded_sequences)"
+    "model_inputs = torch.tensor(encoded_sequences["input_ids"])"
    ]
   },
   {