diff --git a/python/paddle/fluid/tests/unittests/test_dynrnn_gradient_check.py b/python/paddle/fluid/tests/unittests/test_dynrnn_gradient_check.py
index 5b204a1ed6e6a9fadd3f274ad543615c2ebfbb73..7756885166c88eadb77c2c6d56aab767015abc51 100644
--- a/python/paddle/fluid/tests/unittests/test_dynrnn_gradient_check.py
+++ b/python/paddle/fluid/tests/unittests/test_dynrnn_gradient_check.py
@@ -30,12 +30,12 @@ class Memory(object):
         assert val.dtype == self.ex.dtype
         self.cur = val
 
-    def __next__(self):
+    def next(self):
         self.ex = self.cur
         self.cur = None
 
     def __next__(self):
-        next(self)
+        self.next()
 
     def reset(self):
         self.ex = numpy.zeros(shape=self.ex.shape, dtype=self.ex.dtype)