diff --git a/client/python/test/test_client.py b/client/python/test/test_client.py
index b37b524129231..08392e4f5e641 100644
--- a/client/python/test/test_client.py
+++ b/client/python/test/test_client.py
@@ -183,11 +183,13 @@ def my_function(x, progress=gr.Progress()):
             all_progress_data = [
                 p for s in statuses if s.progress_data for p in s.progress_data
             ]
+            count = 0
             for i in range(20):
                 unit = ProgressUnit(
                     index=i, length=20, unit="steps", progress=None, desc=None
                 )
-                assert unit in all_progress_data
+                count += unit in all_progress_data
+            assert count
         finally:
             demo.close()