diff --git a/arxiv-style/main.tex b/arxiv-style/main.tex
index 8f2f127..efcd0a3 100644
--- a/arxiv-style/main.tex
+++ b/arxiv-style/main.tex
@@ -111,7 +111,7 @@ A key empirical and methodological tension in ICS synthesis is that temporal rea
 \begin{figure}[htbp]
   \centering
   \includegraphics[width=0.8\textwidth]{fig-design-v4.png}
-  % \caption{Description of the figure.}
+  \caption{Masked-DDPM: Unified Synthesis for ICS traffic}
   \label{fig:design}
 \end{figure}