diff --git a/pkg/utils/apiutil/serverapi/middleware.go b/pkg/utils/apiutil/serverapi/middleware.go
index 0718702b5a5..9af8d234b34 100644
--- a/pkg/utils/apiutil/serverapi/middleware.go
+++ b/pkg/utils/apiutil/serverapi/middleware.go
@@ -270,7 +270,7 @@ func (h *redirector) waitForLeader(r *http.Request) (leader *pdpb.Member) {
 			}
 		case <-r.Context().Done():
 			return
-		case <-h.s.LoopContext().Done():
+		case <-h.s.Context().Done():
 			return
 		}
 	}