diff --git a/src/mcmcdiags_vis.jl b/src/mcmcdiags_vis.jl
index 84e6e96..7eec2c8 100644
--- a/src/mcmcdiags_vis.jl
+++ b/src/mcmcdiags_vis.jl
@@ -82,7 +82,7 @@ function mcmcdiags_vis(;
                             size = (1500, 1500))
         display(final_plot)
         if save_plots
-            save_plots_to_docs(plt, "mcmc_diagnosis_plots")
+            save_plots_to_docs(final_plot, "mcmc_diagnosis_plots")
         end
     else
         println("NO MCMC DIAGNOSIS PLOTS TO DISPLAY!!!")