diff --git a/src/ainotes.cls b/src/ainotes.cls
index 7048521..9083520 100644
--- a/src/ainotes.cls
+++ b/src/ainotes.cls
@@ -14,7 +14,7 @@
 \usepackage{marginnote}
 \usepackage[bottom]{footmisc}
 \usepackage{scrlayer-scrpage}
-\usepackage{algorithm, listings}
+\usepackage{scrhack, algorithm, listings}
 \usepackage{array, makecell}
 \geometry{ margin=3cm, lmargin=1.5cm, rmargin=4.5cm, marginparwidth=3cm }
 \hypersetup{ colorlinks, citecolor=black, filecolor=black, linkcolor=black, urlcolor=black, linktoc=all }
diff --git a/src/statistical-and-mathematical-methods-for-ai/sections/_linear_algebra.tex b/src/statistical-and-mathematical-methods-for-ai/sections/_linear_algebra.tex
index 746e2b0..633cb5b 100644
--- a/src/statistical-and-mathematical-methods-for-ai/sections/_linear_algebra.tex
+++ b/src/statistical-and-mathematical-methods-for-ai/sections/_linear_algebra.tex
@@ -232,7 +232,7 @@ Common norms are:
         
         The vector $\vec{w} \in U^\perp$ s.t. $\Vert \vec{w} \Vert = 1$ is the \textbf{normal vector} of $U$. \marginnote{Normal vector}
         %
-        \begin{figure}[h]
+        \begin{figure}[ht]
             \centering
             \includegraphics[width=0.4\textwidth]{img/_orthogonal_complement.pdf}
             \caption{Orthogonal complement of a subspace $U \subseteq \mathbb{R}^3$}