diff --git a/internal/datanode/flow_graph_dd_node.go b/internal/datanode/flow_graph_dd_node.go
index 52df9ead7a..ae28faa69e 100644
--- a/internal/datanode/flow_graph_dd_node.go
+++ b/internal/datanode/flow_graph_dd_node.go
@@ -96,7 +96,7 @@ func (ddn *ddNode) Operate(in []Msg) []Msg {
 					continue
 				}
 			}
-			iMsg.insertMessages = append(iMsg.insertMessages, msg.(*msgstream.InsertMsg))
+			iMsg.insertMessages = append(iMsg.insertMessages, imsg)
 		}
 	}