diff --git a/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala
index 5cbb0654eb376..16522e981a338 100644
--- a/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala
+++ b/core/src/test/scala/org/apache/spark/scheduler/DAGSchedulerSuite.scala
@@ -5827,6 +5827,7 @@ class DAGSchedulerSuite extends SparkFunSuite with TempLocalSparkContext with Ti
       // Map stage completes successfully,
       completeShuffleMapStageSuccessfully(0, 0, 3, Seq("hostA", "hostB"))
       taskIdCount += 2
+      sc.listenerBus.waitUntilEmpty()
       assert(completedStage === List(0))
 
       // Now submit the first reducer stage