diff --git a/tutorials/img_cls/test_pt.py b/tutorials/img_cls/test_pt.py
index 7858cca..b16eb63 100644
--- a/tutorials/img_cls/test_pt.py
+++ b/tutorials/img_cls/test_pt.py
@@ -28,7 +28,7 @@
 
 
     ## inference
-    images, labels = dataiter.next()
+    images, labels = next(dataiter)
     print('Ground-truth: ', ' '.join('%5s' % classes[labels[j]] for j in range(4)))
 
     outputs = model(images)