diff --git a/extension/spark-doris-connector/src/main/scala/org/apache/doris/spark/sql/DorisSourceProvider.scala b/extension/spark-doris-connector/src/main/scala/org/apache/doris/spark/sql/DorisSourceProvider.scala
index 774a29c2ba..d8f951b262 100644
--- a/extension/spark-doris-connector/src/main/scala/org/apache/doris/spark/sql/DorisSourceProvider.scala
+++ b/extension/spark-doris-connector/src/main/scala/org/apache/doris/spark/sql/DorisSourceProvider.scala
@@ -63,7 +63,7 @@ private[sql] class DorisSourceProvider extends DataSourceRegister with RelationP
     val maxRowCount = dorisWriterOption.maxRowCount
     val maxRetryTimes = dorisWriterOption.maxRetryTimes
 
-    data.foreachPartition(partition => {
+    data.rdd.foreachPartition(partition => {
 
       val buffer = ListBuffer[String]()
       partition.foreach(row => {