diff --git a/internal/datacoord/server.go b/internal/datacoord/server.go
index 4605f10ca1..6ee0eb1f1b 100644
--- a/internal/datacoord/server.go
+++ b/internal/datacoord/server.go
@@ -277,7 +277,7 @@ func (s *Server) Register() error {
 	metrics.NumNodes.WithLabelValues(fmt.Sprint(paramtable.GetNodeID()), typeutil.DataCoordRole).Inc()
 	log.Info("DataCoord Register Finished")
 
-	s.session.LivenessCheck(s.serverLoopCtx, func() {
+	s.session.LivenessCheck(s.ctx, func() {
 		logutil.Logger(s.ctx).Error("disconnected from etcd and exited", zap.Int64("serverID", s.session.GetServerID()))
 		os.Exit(1)
 	})