diff --git a/ch05/01_main-chapter-code/ch05.ipynb b/ch05/01_main-chapter-code/ch05.ipynb
index 1bf4a22..d8a06ef 100644
--- a/ch05/01_main-chapter-code/ch05.ipynb
+++ b/ch05/01_main-chapter-code/ch05.ipynb
@@ -390,32 +390,6 @@
     "print(\"Token IDs:\\n\", token_ids)"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "id": "f08e4efc-e897-4d35-91f7-9a77b6dde953",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Token IDs:\n",
-      " tensor([[[16657],\n",
-      "         [  339],\n",
-      "         [42826]],\n",
-      "\n",
-      "        [[49906],\n",
-      "         [29669],\n",
-      "         [41751]]])\n"
-     ]
-    }
-   ],
-   "source": [
-    "token_ids = torch.argmax(probas, dim=-1, keepdim=True)\n",
-    "print(\"Token IDs:\\n\", token_ids)"
-   ]
-  },
   {
    "cell_type": "markdown",
    "id": "cee4072c-21ed-4df7-8721-dd2535362573",
@@ -2480,7 +2454,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.4"
+   "version": "3.10.6"
   }
  },
  "nbformat": 4,