diff --git a/CLAUDE.md b/CLAUDE.md
index 6150cf20..235728ac 100644
--- a/CLAUDE.md
+++ b/CLAUDE.md
@@ -41,10 +41,10 @@ black autogalaxy/
 
 ### Plot Output Mode
 
-Set `PYAUTOARRAY_OUTPUT_MODE=1` to capture every figure produced by a script into numbered PNG files in `./output_mode/<script_name>/`. This is useful for visually inspecting all plots from an integration test without needing a display.
+Set `PYAUTO_OUTPUT_MODE=1` to capture every figure produced by a script into numbered PNG files in `./output_mode/<script_name>/`. This is useful for visually inspecting all plots from an integration test without needing a display.
 
 ```bash
-PYAUTOARRAY_OUTPUT_MODE=1 python scripts/my_script.py
+PYAUTO_OUTPUT_MODE=1 python scripts/my_script.py
 # -> ./output_mode/my_script/0_fit.png, 1_tracer.png, ...
 ```
 
diff --git a/autogalaxy/analysis/model_util.py b/autogalaxy/analysis/model_util.py
index 552a38af..2974a54c 100644
--- a/autogalaxy/analysis/model_util.py
+++ b/autogalaxy/analysis/model_util.py
@@ -90,7 +90,7 @@ def mge_model_from(
 
     import os
 
-    if os.environ.get("PYAUTO_WORKSPACE_SMALL_DATASETS") == "1":
+    if os.environ.get("PYAUTO_SMALL_DATASETS") == "1":
         total_gaussians = 2
         gaussian_per_basis = 1
 
diff --git a/autogalaxy/quantity/model/visualizer.py b/autogalaxy/quantity/model/visualizer.py
index 09990786..37838bc3 100644
--- a/autogalaxy/quantity/model/visualizer.py
+++ b/autogalaxy/quantity/model/visualizer.py
@@ -1,7 +1,7 @@
 import os
 
 import autofit as af
-from autoconf.test_mode import is_test_mode
+from autoconf.test_mode import skip_visualization
 
 from autogalaxy.quantity.model.plotter import PlotterQuantity
 
@@ -66,7 +66,7 @@ def visualize(
             via a non-linear search).
         """
 
-        if is_test_mode():
+        if skip_visualization():
             return
 
         fit = analysis.fit_quantity_for_instance(instance=instance)