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 9f691094b..3977e5d54 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
@@ -41,7 +41,7 @@ public class PushState {
   public PushState(CelebornConf conf) {
     pushBufferMaxSize = conf.clientPushBufferMaxSize();
     inFlightRequestTracker = new InFlightRequestTracker(conf, this);
-    failedBatchMap = new ConcurrentHashMap<>();
+    failedBatchMap = JavaUtils.newConcurrentHashMap();
   }
 
   public void cleanup() {