diff --git a/internal/datanode/data_node_test.go b/internal/datanode/data_node_test.go
index d96389304b..f74c9bf21c 100644
--- a/internal/datanode/data_node_test.go
+++ b/internal/datanode/data_node_test.go
@@ -185,6 +185,7 @@ func TestDataNode(t *testing.T) {
 	t.Run("Test BackGroundGC", func(t *testing.T) {
 		ctx, cancel := context.WithCancel(context.Background())
 		node := newIDLEDataNodeMock(ctx, schemapb.DataType_Int64)
+		defer node.Stop()
 
 		vchanNameCh := make(chan string)
 		node.clearSignal = vchanNameCh
diff --git a/internal/datanode/flow_graph_manager_test.go b/internal/datanode/flow_graph_manager_test.go
index cc6c5e2f39..28d6e74f9b 100644
--- a/internal/datanode/flow_graph_manager_test.go
+++ b/internal/datanode/flow_graph_manager_test.go
@@ -45,6 +45,7 @@ func TestFlowGraphManager(t *testing.T) {
 	defer etcdCli.Close()
 
 	node := newIDLEDataNodeMock(ctx, schemapb.DataType_Int64)
+	defer node.Stop()
 	node.SetEtcdClient(etcdCli)
 	err = node.Init()
 	require.Nil(t, err)