diff --git a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb
index f003c5f4a4..b6591c9c21 100644
--- a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb	
+++ b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb	
@@ -105,7 +105,7 @@
    "outputs": [],
    "source": [
     "dataiter = iter(trainloader)\n",
-    "images, labels = dataiter.next()\n",
+    "images, labels = next(dataiter)\n",
     "print(type(images))\n",
     "print(images.shape)\n",
     "print(labels.shape)"