diff --git a/backends-clickhouse/src/main/scala/io/glutenproject/backendsapi/clickhouse/CHIteratorApi.scala b/backends-clickhouse/src/main/scala/io/glutenproject/backendsapi/clickhouse/CHIteratorApi.scala
index 7fa21e413435..1d97f89114c1 100644
--- a/backends-clickhouse/src/main/scala/io/glutenproject/backendsapi/clickhouse/CHIteratorApi.scala
+++ b/backends-clickhouse/src/main/scala/io/glutenproject/backendsapi/clickhouse/CHIteratorApi.scala
@@ -89,7 +89,7 @@ class CHIteratorApi extends IteratorApi with Logging with LogLevelUtil {
                 lengths,
                 partitionColumns.map(_.asJava).asJava,
                 fileFormats(i)),
-              SoftAffinityUtil.getFilePartitionLocations(f))
+              SoftAffinityUtil.getFilePartitionLocations(f, f.files))
           case _ =>
             throw new UnsupportedOperationException(s"Unsupport operators.")
         })