diff --git a/intro-to-pytorch/Part 5 - Inference and Validation (Solution).ipynb b/intro-to-pytorch/Part 5 - Inference and Validation (Solution).ipynb
index 117bb5e3c9..8822969466 100644
--- a/intro-to-pytorch/Part 5 - Inference and Validation (Solution).ipynb	
+++ b/intro-to-pytorch/Part 5 - Inference and Validation (Solution).ipynb	
@@ -426,8 +426,8 @@
     "        test_losses.append(test_loss/len(testloader))\n",
     "\n",
     "        print(\"Epoch: {}/{}.. \".format(e+1, epochs),\n",
-    "              \"Training Loss: {:.3f}.. \".format(running_loss/len(trainloader)),\n",
-    "              \"Test Loss: {:.3f}.. \".format(test_loss/len(testloader)),\n",
+    "              \"Training Loss: {:.3f}.. \".format(train_losses[-1]),\n",
+    "              \"Test Loss: {:.3f}.. \".format(test_losses[-1]),\n",
     "              \"Test Accuracy: {:.3f}\".format(accuracy/len(testloader)))"
    ]
   },