diff --git a/paimon-flink/paimon-flink-common/src/main/java/org/apache/paimon/flink/sink/StoreCommitter.java b/paimon-flink/paimon-flink-common/src/main/java/org/apache/paimon/flink/sink/StoreCommitter.java
index b3e74a3f6125..9ae53f401325 100644
--- a/paimon-flink/paimon-flink-common/src/main/java/org/apache/paimon/flink/sink/StoreCommitter.java
+++ b/paimon-flink/paimon-flink-common/src/main/java/org/apache/paimon/flink/sink/StoreCommitter.java
@@ -146,6 +146,9 @@ public Map<Long, List<Committable>> groupByCheckpoint(Collection<Committable> co
     @Override
     public void close() throws Exception {
         commit.close();
+        if (partitionMarkDone != null) {
+            partitionMarkDone.close();
+        }
     }
 
     private void calcNumBytesAndRecordsOut(List<ManifestCommittable> committables) {