diff --git a/internal/datanode/mock_test.go b/internal/datanode/mock_test.go
index f72bbb5f1d..e1b54617d8 100644
--- a/internal/datanode/mock_test.go
+++ b/internal/datanode/mock_test.go
@@ -51,10 +51,10 @@ func newIDLEDataNodeMock(ctx context.Context) *DataNode {
 		collectionID:   1,
 		collectionName: "collection-1",
 	}
-	node.SetRootCoordInterface(rc)
+	node.rootCoord = rc
 
 	ds := &DataCoordFactory{}
-	node.SetDataCoordInterface(ds)
+	node.dataCoord = ds
 
 	return node
 }
@@ -82,11 +82,10 @@ func newHEALTHDataNodeMock(dmChannelName string) *DataNode {
 		collectionID:   1,
 		collectionName: "collection-1",
 	}
-
-	node.SetRootCoordInterface(ms)
+	node.rootCoord = ms
 
 	ds := &DataCoordFactory{}
-	node.SetDataCoordInterface(ds)
+	node.dataCoord = ds
 
 	return node
 }