diff --git a/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb b/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
index 351ef703c..2d70a13d5 100644
--- a/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
+++ b/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
@@ -119,7 +119,7 @@
     "!python -m pip install pydicom\n",
     "\n",
     "# Install tcia_utils to download the datasets.\n",
-    "!python -m pip install --upgrade -q tcia_utils\n",
+    "!python -m pip install --upgrade -q tcia_utils==3.2.1\n",
     "\n",
     "\n",
     "# This is the installation required for itkWidgets.\n",
@@ -245,7 +245,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {
     "id": "7wR4HJu-emJP"
    },
@@ -254,7 +254,7 @@
     "dicom_data_dir = \"tciaDownload\"\n",
     "\n",
     "# The series_uid defines their directory where the MR data was stored on disk.\n",
-    "mr_series_uid = df.at[df.Modality.eq(\"MR\").idxmax(), \"Series ID\"]\n",
+    "mr_series_uid = df.at[df.Modality.eq(\"MR\").idxmax(), \"SeriesInstanceUID\"]\n",
     "mr_dir = os.path.join(dicom_data_dir, mr_series_uid)\n",
     "\n",
     "# Read the DICOM MR series' objects and reconstruct them into a 3D ITK image.\n",
@@ -272,7 +272,7 @@
    "outputs": [],
    "source": [
     "# The series_uid defines where the SEG was stored on disk.  It is stored in a single file.\n",
-    "seg_series_uid = df.at[df.Modality.eq(\"SEG\").idxmax(), \"Series ID\"]\n",
+    "seg_series_uid = df.at[df.Modality.eq(\"SEG\").idxmax(), \"SeriesInstanceUID\"]\n",
     "seg_dir = os.path.join(dicom_data_dir, seg_series_uid)\n",
     "seg_file = glob.glob(os.path.join(seg_dir, \"*.dcm\"))[0]\n",
     "\n",
@@ -730,9 +730,7 @@
     },
     {
      "data": {
-      "application/javascript": [
-       "window.connectPlugin && window.connectPlugin(\"7ba53bd6-ca11-4e56-a0dd-f80199194b50\")"
-      ],
+      "application/javascript": "window.connectPlugin && window.connectPlugin(\"7ba53bd6-ca11-4e56-a0dd-f80199194b50\")",
       "text/plain": [
        "<IPython.core.display.Javascript object>"
       ]