diff --git a/client-spark/spark-3-4/src/main/java/org/apache/spark/shuffle/celeborn/BasedShuffleWriter.java b/client-spark/spark-3-4/src/main/java/org/apache/spark/shuffle/celeborn/BasedShuffleWriter.java
index f732a68ad43..e3bbc2381de 100644
--- a/client-spark/spark-3-4/src/main/java/org/apache/spark/shuffle/celeborn/BasedShuffleWriter.java
+++ b/client-spark/spark-3-4/src/main/java/org/apache/spark/shuffle/celeborn/BasedShuffleWriter.java
@@ -199,10 +199,9 @@ protected void pushGiantRecord(int partitionId, byte[] buffer, int numBytes) thr
   }
 
   /**
-   * This method will push the remaining data and close these pushers.
-   * It's important, will send Mapper End RPC to LifecycleManager to update
-   * the attempt of the corresponding task.
-   * We should only call this method when the task is successfully completed.
+   * This method will push the remaining data and close these pushers. It's important, will send
+   * Mapper End RPC to LifecycleManager to update the attempt of the corresponding task. We should
+   * only call this method when the task is successfully completed.
    */
   protected void close() throws IOException, InterruptedException {
     long pushMergedDataTime = System.nanoTime();