diff --git a/connectors/flink/src/test/java/io/delta/flink/sink/DeltaSinkStreamingExecutionITCase.java b/connectors/flink/src/test/java/io/delta/flink/sink/DeltaSinkStreamingExecutionITCase.java
index 330936168a0..ff1c82da671 100644
--- a/connectors/flink/src/test/java/io/delta/flink/sink/DeltaSinkStreamingExecutionITCase.java
+++ b/connectors/flink/src/test/java/io/delta/flink/sink/DeltaSinkStreamingExecutionITCase.java
@@ -468,7 +468,7 @@ private List<String> getDeltaCheckpointFiles(String deltaTablePath) throws IOExc
             return stream
                 .filter(file -> !Files.isDirectory(file))
                 .map(file -> file.getFileName().toString())
-                .filter(fileName -> !fileName.endsWith(".json"))
+                .filter(fileName -> fileName.endsWith(".checkpoint.parquet"))
                 .collect(Collectors.toList());
         }
     }