diff --git a/python/pyspark/sql/tests/streaming/test_streaming_listener.py b/python/pyspark/sql/tests/streaming/test_streaming_listener.py
index 1f5b0f573807a..c3ae62e64cc30 100644
--- a/python/pyspark/sql/tests/streaming/test_streaming_listener.py
+++ b/python/pyspark/sql/tests/streaming/test_streaming_listener.py
@@ -381,8 +381,7 @@ def verify(test_listener):
                     .start()
                 )
                 self.assertTrue(q.isActive)
-                while progress_event is None or progress_event.batchId == 0:
-                    q.awaitTermination(0.5)
+                q.awaitTermination(10)
                 q.stop()
 
                 # Make sure all events are empty