diff --git a/python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py b/python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py
index 110460497070cff2e597bf26117fd72dccbf5444..41fa39e06be8c9ce4da546adc6ac740f6a90db51 100644
--- a/python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py
+++ b/python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py
@@ -175,8 +175,8 @@ class TestAsyncSSAGraphExecutor(unittest.TestCase):
                 with fluid.program_guard(
                         fluid.Program(), startup_program=fluid.Program()):
                     test()
-        assert step_list[0] / 2 == step_list[1]
-        assert step_list[1] / 2 == step_list[2]
+        assert int(step_list[0] / 2) == int(step_list[1])
+        assert int(step_list[1] / 2) == int(step_list[2])
 
 
 if __name__ == "__main__":