diff --git a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb
index 1f4ad1ee49..903a9ffc3d 100644
--- a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb	
+++ b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb	
@@ -172,7 +172,7 @@
     "\n",
     "h = activation(torch.mm(inputs, w1) + b1)\n",
     "\n",
-    "out = torch.mm(h, w2) + b2"
+    "out = activation(torch.mm(h, w2) + b2)"
    ]
   },
   {