\begin{figure}[!ht]
    \centering
    \begin{tikzpicture}[node distance=3cm]
    \node (thm-uniform-consistency) [block] {\Cref{thm:uniform-consistency-squared}} ;
    \node (lemma-expectation-close) [block, below of=thm-uniform-consistency] {\Cref{lemma:expectation-close}} ;
    \node (lemma-projection-effects) [block, right of=lemma-expectation-close] {\Cref{lemma:projection-effects}} ;
    \node (lemma-cRk) [block, right of=lemma-projection-effects] {\Cref{lemma:cRk}} ;
    % \node (lemma-Rloo) [block, right of=lemma-cRk, node distance=3cm] {\Cref{lemma:Rloo}} ;
    \node (lemma-Rloo) [block, right of=lemma-cRk, node distance=3cm] {\Cref{lemma:Rloo}} ;
    \node (lemma-gradient-upper-bound) [block, below of=lemma-Rloo, node distance=3cm] {\Cref{lemma:gradient-upper-bound}} ;
    \node (lemma-concentration) [block, right of=lemma-gradient-upper-bound] {\Cref{lemma:concentration}} ;
    \node (sec-proof-lemma-gradient-upper-bound) [block, below of=lemma-gradient-upper-bound, node distance=3cm] {\Cref{sec:proof-lemma:gradient-upper-bound}} ;
    \node (lemma-norm-theta) [block, below of=lemma-cRk] {\Cref{lemma:norm-theta}} ;
    \node (lemma-op-subGaussian) [block, below of=lemma-projection-effects] {\Cref{lemma:op-subGaussian}} ;
    \node (lemma-op-Sigma) [block, below of=lemma-op-subGaussian] {\Cref{lemma:op-Sigma}} ;
    \node (prop-T2-prop-Gozlan) [block, left of=lemma-op-Sigma, node distance = 4cm] {\Cref{prop:Gozlan}} ;
    %\node (lemma-norm-y) [block, below of=lemma-norm-theta] {\Cref{lemma:norm-y}} ;
    \draw [arrow] (lemma-Rloo) -- (thm-uniform-consistency) ;
    \draw [arrow] (lemma-cRk) -- (thm-uniform-consistency) ;
    \draw [arrow] (lemma-projection-effects) -- (thm-uniform-consistency) ;
    \draw [arrow] (lemma-expectation-close) -- (thm-uniform-consistency) ;
    \draw [arrow] (lemma-gradient-upper-bound) -- (lemma-Rloo) ;
    \draw [arrow] (sec-proof-lemma-gradient-upper-bound) -- (lemma-gradient-upper-bound) ;
    \draw [arrow] (lemma-concentration) -- (lemma-Rloo) ;
    \draw [arrow] (sec-proof-lemma-gradient-upper-bound) -- (lemma-cRk) ;
    \draw [arrow] (lemma-norm-theta) -- (lemma-expectation-close);
    \draw [arrow] (lemma-norm-theta) -- (lemma-cRk) ;
    \draw [arrow] (lemma-op-subGaussian) -- (lemma-projection-effects) ;
    \draw [arrow] (lemma-op-subGaussian) -- (lemma-expectation-close) ;
    \draw [arrow] (lemma-op-Sigma) -- (lemma-op-subGaussian) ;
    %\draw [arrow] (lemma-norm-y) -- (lemma-norm-theta) ;
    \draw [arrow] (prop-T2-prop-Gozlan) -- (lemma-op-Sigma) ;
    \end{tikzpicture}
    \caption{Schematic for the proof of \Cref{thm:uniform-consistency-squared}
    }
    \label{fig:schematic-proof-thm:uniform-consistency}
\end{figure}