diff --git a/internal/rootcoord/root_coord_test.go b/internal/rootcoord/root_coord_test.go
index 583f38e7a0..14d84c0738 100644
--- a/internal/rootcoord/root_coord_test.go
+++ b/internal/rootcoord/root_coord_test.go
@@ -1844,7 +1844,7 @@ func TestRootCoord2(t *testing.T) {
 
 		pChan := core.MetaTable.ListCollectionPhysicalChannels()
 		dmlStream, _ := msFactory.NewMsgStream(ctx)
-		dmlStream.AsConsumer(pChan, Params.MsgChannelSubName)
+		dmlStream.AsConsumer([]string{pChan[0]}, Params.MsgChannelSubName)
 		dmlStream.Start()
 
 		msgs := getNotTtMsg(ctx, 2, dmlStream.Chan())