diff --git a/flink-end-to-end-tests/flink-sql-client-test/src/test/java/SqlClientITCase.java b/flink-end-to-end-tests/flink-sql-client-test/src/test/java/SqlClientITCase.java
index de56f79491ff5d..f37528480f22e3 100644
--- a/flink-end-to-end-tests/flink-sql-client-test/src/test/java/SqlClientITCase.java
+++ b/flink-end-to-end-tests/flink-sql-client-test/src/test/java/SqlClientITCase.java
@@ -290,10 +290,15 @@ private void verifyNumberOfResultRecords(String resultFilePath, int expectedNumb
         File tempOutputFile = new File(tempDir, "records.out");
         String tempOutputFilepath = tempOutputFile.toString();
         GenericContainer<?> taskManager = flink.getTaskManagers().get(0);
-        Thread.sleep(5000); // prevent NotFoundException: Status 404
-        taskManager.copyFileFromContainer(resultFilePath, tempOutputFilepath);
-
-        int numberOfResultRecords = UpsertTestFileUtil.getNumberOfRecords(tempOutputFile);
+        int numberOfResultRecords;
+        while (true) {
+            Thread.sleep(50); // prevent NotFoundException: Status 404
+            taskManager.copyFileFromContainer(resultFilePath, tempOutputFilepath);
+            numberOfResultRecords = UpsertTestFileUtil.getNumberOfRecords(tempOutputFile);
+            if (numberOfResultRecords == expectedNumberOfRecords) {
+                break;
+            }
+        }
         assertThat(numberOfResultRecords).isEqualTo(expectedNumberOfRecords);
     }