diff --git a/client/src/test/java/org/apache/celeborn/client/write/DataPushQueueSuitJ.java b/client/src/test/java/org/apache/celeborn/client/write/DataPushQueueSuitJ.java
index bbd432bc855..b322fee3928 100644
--- a/client/src/test/java/org/apache/celeborn/client/write/DataPushQueueSuitJ.java
+++ b/client/src/test/java/org/apache/celeborn/client/write/DataPushQueueSuitJ.java
@@ -147,6 +147,7 @@ public void testDataPushQueue() throws Exception {
 
     Thread.sleep(15 * 1000);
     running.set(false);
+    Thread.sleep(15 * 1000);
 
     for (int i = 0; i < numWorker; i++) {
       Assert.assertArrayEquals(workerData.get(i).toArray(), tarWorkerData.get(i).toArray());