diff --git a/code/plot_functions.py b/code/plot_functions.py
index 7bf001f..5fb4c17 100644
--- a/code/plot_functions.py
+++ b/code/plot_functions.py
@@ -136,7 +136,7 @@ def plot_highlighted_integrals(frequency, power, points, color_mapping, points_c
     ax.set_title('Power Spectrum with Highlighted Integrals')
     ax.legend()
     
-    return fig
+    return fig, ax