diff --git a/qemu/tests/balloon_check.py b/qemu/tests/balloon_check.py
index b7eb102df114b854915067bc026ef8ef356b80af..40c35c835b744f7817afd433307db232b67a7791 100644
--- a/qemu/tests/balloon_check.py
+++ b/qemu/tests/balloon_check.py
@@ -174,7 +174,7 @@ class BallooningTest(MemoryBaseTest):
         """
         logging.info("Wait until guest memory don't change")
         is_stable = self._mem_state()
-        utils_misc.wait_for(is_stable.next, timeout, step=10.0)
+        utils_misc.wait_for(lambda: next(is_stable), timeout, step=10.0)
 
     def get_memory_boundary(self, balloon_type=''):
         """