diff --git a/src/yieldplotlib/plots/yip_plots.py b/src/yieldplotlib/plots/yip_plots.py
index e391a3a0..68e5e636 100644
--- a/src/yieldplotlib/plots/yip_plots.py
+++ b/src/yieldplotlib/plots/yip_plots.py
@@ -65,6 +65,7 @@ def update(frame):
     # Save the animation to file.
     ani.save(save_name, fps=2, writer="ffmpeg")
     plt.close(fig)
+    return ani
 
 
 def plot_core_throughtput(