diff --git a/python/pyspark/sql/tests/streaming/test_streaming_listener.py b/python/pyspark/sql/tests/streaming/test_streaming_listener.py
index 30d6eee93879d..d28fb57a0da23 100644
--- a/python/pyspark/sql/tests/streaming/test_streaming_listener.py
+++ b/python/pyspark/sql/tests/streaming/test_streaming_listener.py
@@ -230,7 +230,7 @@ def onQueryTerminated(self, event):
 
             q = observed_ds.writeStream.format("noop").start()
 
-            while q.lastProgress is None or q.lastProgress.batchId == 0:
+            while q.lastProgress is None or q.lastProgress.progress.batchId == 0:
                 q.awaitTermination(0.5)
 
             time.sleep(5)