diff --git a/appendix-A/03_main-chapter-code/code-part1.ipynb b/appendix-A/03_main-chapter-code/code-part1.ipynb
index 39e3f2e..71490fa 100644
--- a/appendix-A/03_main-chapter-code/code-part1.ipynb
+++ b/appendix-A/03_main-chapter-code/code-part1.ipynb
@@ -700,7 +700,7 @@
    ],
    "source": [
     "with torch.no_grad():\n",
-    "    out = F.softmax(model(X), dim=1)\n",
+    "    out = torch.softmax(model(X), dim=1)\n",
     "print(out)"
    ]
   },
@@ -986,7 +986,7 @@
    ],
    "source": [
     "torch.set_printoptions(sci_mode=False)\n",
-    "probas = F.softmax(outputs, dim=1)\n",
+    "probas = torch.softmax(outputs, dim=1)\n",
     "print(probas)\n",
     "\n",
     "predictions = torch.argmax(outputs, dim=1)\n",