diff --git a/internal/datanode/importv2/util.go b/internal/datanode/importv2/util.go
index 545e729b51..16c003b481 100644
--- a/internal/datanode/importv2/util.go
+++ b/internal/datanode/importv2/util.go
@@ -228,13 +228,17 @@ func RunBm25Function(task *ImportTask, data *storage.InsertData) error {
 		if err != nil {
 			return err
 		}
+
 		if runner == nil {
 			continue
 		}
-		inputDatas := make([]any, 0, len(fn.InputFieldIds))
-		for _, inputFieldID := range fn.InputFieldIds {
+
+		inputFieldIDs := lo.Map(runner.GetInputFields(), func(field *schemapb.FieldSchema, _ int) int64 { return field.GetFieldID() })
+		inputDatas := make([]any, 0, len(inputFieldIDs))
+		for _, inputFieldID := range inputFieldIDs {
 			inputDatas = append(inputDatas, data.Data[inputFieldID].GetDataRows())
 		}
+
 		outputFieldData, err := runner.BatchRun(inputDatas...)
 		if err != nil {
 			return err