diff --git a/code/plot_functions.py b/code/plot_functions.py
index b35cd02..aa634b8 100644
--- a/code/plot_functions.py
+++ b/code/plot_functions.py
@@ -104,8 +104,8 @@ def plot_highlighted_integrals(ax, frequency, power, points, color_mapping, poin
     -------
     None
     """
-    _, _, AM, df, eodf, nyquist, stim_freq = u.sam_data(sam)
-
+    _, _, _, df, eodf, nyquist, stim_freq = u.sam_data(sam)
+    AM = u.find_AM(eodf, nyquist, stim_freq)
     # Plot the power spectrum on the provided axes
     ax.plot(frequency, power, color="k")