diff --git a/common/src/main/java/org/apache/celeborn/common/write/PushState.java b/common/src/main/java/org/apache/celeborn/common/write/PushState.java
index f63906224..46714c4e8 100644
--- a/common/src/main/java/org/apache/celeborn/common/write/PushState.java
+++ b/common/src/main/java/org/apache/celeborn/common/write/PushState.java
@@ -36,7 +36,7 @@ public class PushState {
   private final InFlightRequestTracker inFlightRequestTracker;
   // partition id -> CommitMetadata
   private final ConcurrentHashMap<Integer, CommitMetadata> commitMetadataMap =
-      new ConcurrentHashMap<>();
+      JavaUtils.newConcurrentHashMap();
 
   private final Map<String, LocationPushFailedBatches> failedBatchMap;