Fixed the previous commit for Python3.

1 files changed, 3 insertions, 2 deletions

diff --git a/tests/model_fields/models.py b/tests/model_fields/models.py
index 8dfdcb91d5..0e9fe23c67 100644
--- a/tests/model_fields/models.py
+++ b/tests/model_fields/models.py
@@ -10,6 +10,7 @@ except ImportError:
 from django.core.files.storage import FileSystemStorage
 from django.db import models
 from django.db.models.fields.files import ImageFieldFile, ImageField
+from django.utils import six
 
 
 class Foo(models.Model):
@@ -43,14 +44,14 @@ class Whiz(models.Model):
     c = models.IntegerField(choices=CHOICES, null=True)
 
 
-class Counter:
+class Counter(six.Iterator):
     def __init__(self):
         self.n = 1
 
     def __iter__(self):
         return self
 
-    def next(self):  # Python 3: def __next__(self)
+    def __next__(self):
         if self.n > 5:
             raise StopIteration
         else: