diff --git a/preamble.tex b/preamble.tex
index 3c0bc1b..f28eae0 100644
--- a/preamble.tex
+++ b/preamble.tex
@@ -953,7 +953,10 @@
 \newlength{\eqindent}
 \setlength{\eqindent}{1.4em} % Recommended: baselineskip, 1.3em, 1.4em, 1.5em
 
-\AtBeginEnvironment{equation}{\setlength\abovedisplayskip{\eqindent - \baselineskip}}
-\AtBeginEnvironment{equation}{\setlength\belowdisplayskip{\eqindent}}
-\AtBeginEnvironment{equation}{\setlength\abovedisplayshortskip{\eqindent - \baselineskip}}
-\AtBeginEnvironment{equation}{\setlength\belowdisplayshortskip{\eqindent}}
\ No newline at end of file
+% Note: if necessary, supplement the list with the necessary math environment
+\foreach \env in {equation, equation*, align, align*}{
+    \AtBeginEnvironment{\env}{\setlength\abovedisplayskip{\eqindent - \baselineskip}}
+    \AtBeginEnvironment{\env}{\setlength\belowdisplayskip{\eqindent}}
+    \AtBeginEnvironment{\env}{\setlength\abovedisplayshortskip{\eqindent - \baselineskip}}
+    \AtBeginEnvironment{\env}{\setlength\belowdisplayshortskip{\eqindent}}
+}