diff --git a/autogalaxy/plot/__init__.py b/autogalaxy/plot/__init__.py
index 37f3b28f..aa1b3529 100644
--- a/autogalaxy/plot/__init__.py
+++ b/autogalaxy/plot/__init__.py
@@ -1,3 +1,11 @@
+from autofit.non_linear.plot import (
+    corner_cornerpy,
+    corner_anesthetic,
+    subplot_parameters,
+    log_likelihood_vs_iteration,
+    output_figure,
+)
+
 from autogalaxy.plot.plot_utils import plot_array, plot_grid, fits_array
 
 from autoarray.dataset.plot.imaging_plots import (
diff --git a/docs/api/plot.rst b/docs/api/plot.rst
index 10392905..5e9d6b3c 100644
--- a/docs/api/plot.rst
+++ b/docs/api/plot.rst
@@ -71,22 +71,20 @@ Create figures and subplots showing quantities of standard **PyAutoGalaxy** obje
     subplot_fit_ellipse
     subplot_ellipse_errors
 
-Non-linear Search Plotters [aplt]
----------------------------------
+Non-linear Search Plot Functions [aplt]
+---------------------------------------
 
-Create figures and subplots of non-linear search specific visualization of every search algorithm supported
-by **PyAutoGalaxy**.
+Module-level functions for visualizing non-linear search results.
 
-.. currentmodule:: autogalaxy.plot
+.. currentmodule:: autofit.plot
 
 .. autosummary::
    :toctree: _autosummary
-   :template: custom-class-template.rst
-   :recursive:
 
-   NestPlotter
-   MCMCPlotter
-   MLEPlotter
+   corner_cornerpy
+   corner_anesthetic
+   subplot_parameters
+   log_likelihood_vs_iteration
 
 Plot Customization [aplt]
 -------------------------