diff --git a/transformers_doc/en/pytorch/language_modeling.ipynb b/transformers_doc/en/pytorch/language_modeling.ipynb
index dcdd2ada..e41cc768 100644
--- a/transformers_doc/en/pytorch/language_modeling.ipynb
+++ b/transformers_doc/en/pytorch/language_modeling.ipynb
@@ -124,7 +124,7 @@
    "source": [
     "from datasets import load_dataset\n",
     "\n",
-    "eli5 = load_dataset(\"eli5\", split=\"train_asks[:5000]\")"
+    "eli5 = load_dataset(\"eli5_category\", split=\"train[:5000]\")\n"
    ]
   },
   {