diff --git a/notebooks/tutorials/model-training/03-data-scientist-download-results.ipynb b/notebooks/tutorials/model-training/03-data-scientist-download-results.ipynb
index de5c4270d2e..a53cdd57c35 100644
--- a/notebooks/tutorials/model-training/03-data-scientist-download-results.ipynb
+++ b/notebooks/tutorials/model-training/03-data-scientist-download-results.ipynb
@@ -156,7 +156,7 @@
    "outputs": [],
    "source": [
     "# third party\n",
-    "import torch.nn as nn\n",
+    "from torch import nn\n",
     "\n",
     "\n",
     "class MLP(nn.Module):\n",