diff --git a/notebooks/kbmod_visualize.ipynb b/notebooks/kbmod_visualize.ipynb
index bd13f7169..df27b99f9 100644
--- a/notebooks/kbmod_visualize.ipynb
+++ b/notebooks/kbmod_visualize.ipynb
@@ -24,6 +24,7 @@
    "outputs": [],
    "source": [
     "# everything we will need for this demo\n",
+    "from kbmod.data_interface import load_deccam_layered_image\n",
     "import kbmod.search as kb\n",
     "import matplotlib.pyplot as plt\n",
     "import math\n",
@@ -51,7 +52,7 @@
    "outputs": [],
    "source": [
     "p = kb.PSF(1.0)\n",
-    "im = kb.LayeredImage(im_path + \"000000.fits\", p)\n",
+    "im = load_deccam_layered_image(im_path + \"000000.fits\", p)\n",
     "print(f\"Loaded a {im.get_width()} by {im.get_height()} image at time {im.get_obstime()}\")"
    ]
   },
@@ -89,12 +90,10 @@
    "source": [
     "files = [im_path + f for f in os.listdir(im_path) if \".fits\" in f]\n",
     "files.sort()\n",
-    "\n",
-    "# Create default PSFs for each image.\n",
-    "all_psfs = [p for _ in range(len(files))]\n",
+    "imgs = [load_deccam_layered_image(f, p) for f in files]\n",
     "\n",
     "# Load the images.\n",
-    "stack = kb.ImageStack(files, all_psfs)\n",
+    "stack = kb.ImageStack(imgs)\n",
     "\n",
     "num_images = stack.img_count()\n",
     "print(f\"Loaded {num_images} images.\")"
@@ -216,9 +215,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python (.conda-kbmod)",
+   "display_name": "Jeremy's KBMOD",
    "language": "python",
-   "name": "conda-env-.conda-kbmod-py"
+   "name": "kbmod_jk"
   },
   "language_info": {
    "codemirror_mode": {