diff --git a/pkg/conn/conn.go b/pkg/conn/conn.go
index 20398c639..fed35a099 100644
--- a/pkg/conn/conn.go
+++ b/pkg/conn/conn.go
@@ -191,8 +191,8 @@ func NewMgr(
 		grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(maxMsgSize)),
 		grpc.WithDefaultCallOptions(grpc.MaxCallSendMsgSize(maxMsgSize)),
 	}
-	pdClient, err := pd.NewClient(
-		addrs, securityOption, pd.WithGRPCDialOptions(maxCallMsgSize...))
+	pdClient, err := pd.NewClientWithContext(
+		ctx, addrs, securityOption, pd.WithGRPCDialOptions(maxCallMsgSize...))
 	if err != nil {
 		log.Error("fail to create pd client", zap.Error(err))
 		return nil, err