diff --git a/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/partitioner/BucketAssigner.java b/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/partitioner/BucketAssigner.java
index e73890f3b54b..ebb47a91eb57 100644
--- a/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/partitioner/BucketAssigner.java
+++ b/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/partitioner/BucketAssigner.java
@@ -29,7 +29,6 @@
 
 import org.apache.flink.annotation.VisibleForTesting;
 import org.apache.flink.runtime.state.KeyGroupRangeAssignment;
-import org.apache.flink.util.Preconditions;
 import org.apache.log4j.LogManager;
 import org.apache.log4j.Logger;
 
@@ -309,9 +308,6 @@ public boolean canAssign() {
      * Remembers to invoke {@link #canAssign()} first.
      */
     public void assign() {
-      Preconditions.checkState(canAssign(),
-          "Can not assign insert to small file: assigned => "
-              + this.assigned + " totalUnassigned => " + this.totalUnassigned);
       this.assigned++;
     }
   }
@@ -339,9 +335,6 @@ public boolean canAssign() {
      * Remembers to invoke {@link #canAssign()} first.
      */
     public void assign() {
-      Preconditions.checkState(canAssign(),
-          "Can not assign insert to new file: assigned => "
-              + this.assigned + " totalUnassigned => " + this.totalUnassigned);
       this.assigned++;
     }
   }