diff --git a/internal/storage/utils.go b/internal/storage/utils.go
index ae34cd4242..d00d916db3 100644
--- a/internal/storage/utils.go
+++ b/internal/storage/utils.go
@@ -618,7 +618,8 @@ func ColumnBasedInsertMsgToInsertData(msg *msgstream.InsertMsg, collSchema *sche
 			srcData := srcFields[field.FieldID].GetScalars().GetArrayData().GetData()
 
 			fieldData := &ArrayFieldData{
-				Data: make([]*schemapb.ScalarField, 0, len(srcData)),
+				ElementType: field.GetElementType(),
+				Data:        make([]*schemapb.ScalarField, 0, len(srcData)),
 			}
 
 			fieldData.Data = append(fieldData.Data, srcData...)
@@ -736,7 +737,8 @@ func mergeStringField(data *InsertData, fid FieldID, field *StringFieldData) {
 func mergeArrayField(data *InsertData, fid FieldID, field *ArrayFieldData) {
 	if _, ok := data.Data[fid]; !ok {
 		fieldData := &ArrayFieldData{
-			Data: nil,
+			ElementType: field.ElementType,
+			Data:        nil,
 		}
 		data.Data[fid] = fieldData
 	}