From 3cb6247271d2416b6d1ff9314754a3578540822e Mon Sep 17 00:00:00 2001
From: Liangliang Gu <marsishandsome@gmail.com>
Date: Mon, 14 Feb 2022 17:43:10 +0800
Subject: [PATCH] cherry pick #458 to release-3.1

Signed-off-by: ti-srebot <ti-srebot@pingcap.com>
---
 .../java/org/tikv/common/region/AbstractRegionStoreClient.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java b/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
index 7bccbb77d78..6b33a9a97c4 100644
--- a/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
+++ b/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
@@ -316,7 +316,7 @@ private TiStore switchProxyStore(BackOffer backOffer) {
       header.put(TiConfiguration.FORWARD_META_DATA_KEY, store.getStore().getAddress());
       Kvrpcpb.RawGetRequest rawGetRequest =
           Kvrpcpb.RawGetRequest.newBuilder()
-              .setContext(region.getReplicaContext(peer))
+              .setContext(region.getReplicaContext(region.getLeader()))
               .setKey(key)
               .build();
       ListenableFuture<Kvrpcpb.RawGetResponse> task =