% \begin{wrapfigure}{r}{0.2\textwidth}
\begin{figure}[t!]
% \vspace{-2mm}
% \begin{minipage}{\linewidth}
\centering
% \captionsetup[subfigure]{justification=centering}
\hfill\subfigure[DoorKey]{\includegraphics[width=0.38\linewidth]{figure/source/environments/MiniGrid-Color-DoorKey-Random-Goal-12x12-C5-v0.png}
\label{fig:environments_colored_doorkey}}\hfill
% \hfill
% \subcaption{DoorKey}
% \centering
% }
% \hfill
\subfigure[Sokoban]{\includegraphics[width=0.38\linewidth]{figure/source/environments/Sokoban-PushAndPull-7x7-B1-C5.png}
\label{fig:environments_colored_sokoban}}\hfill\null
% \end{minipage}
% }
\vspace{-5pt}
\caption{{Sample images for each environment.}}
\label{fig:environments}
\vspace{-5pt}
\end{figure}
% \end{wrapfigure}