diff --git a/notebooks/interferometer/fit.ipynb b/notebooks/interferometer/fit.ipynb
index fea5768a..c220331f 100644
--- a/notebooks/interferometer/fit.ipynb
+++ b/notebooks/interferometer/fit.ipynb
@@ -527,8 +527,10 @@
       "metadata": {},
       "source": [
         "galaxy_model_image = fit.galaxy_image_dict[galaxy]\n",
-        "galaxy_model_image.output_to_fits(\n",
-        "    file_path=dataset_path / \"galaxy_model_image.fits\", overwrite=True\n",
+        "aplt.fits_array(\n",
+        "    array=galaxy_model_image,\n",
+        "    file_path=dataset_path / \"galaxy_model_image.fits\",\n",
+        "    overwrite=True,\n",
         ")"
       ],
       "outputs": [],
diff --git a/scripts/interferometer/fit.py b/scripts/interferometer/fit.py
index caf4dcf6..a477d8b9 100644
--- a/scripts/interferometer/fit.py
+++ b/scripts/interferometer/fit.py
@@ -300,8 +300,10 @@
 we could fit this image again with an independent pipeline.
 """
 galaxy_model_image = fit.galaxy_image_dict[galaxy]
-galaxy_model_image.output_to_fits(
-    file_path=dataset_path / "galaxy_model_image.fits", overwrite=True
+aplt.fits_array(
+    array=galaxy_model_image,
+    file_path=dataset_path / "galaxy_model_image.fits",
+    overwrite=True,
 )
 
 """