diff --git a/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb b/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
index f1dc6f0e5..a9fc2da09 100644
--- a/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
+++ b/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
@@ -116,7 +116,7 @@
     "!python -c \"import cv2\" || pip install -q opencv-python-headless\n",
     "\n",
     "# These are the libraries used to read DICOM Seg objects.\n",
-    "!python -m pip install -q pydicom pydicom-seg\n",
+    "!python -m pip install -q pydicom==2.4.4 pydicom-seg\n",
     "\n",
     "# Install tcia_utils to download the datasets.\n",
     "!python -m pip install --upgrade -q --no-deps tcia_utils\n",
@@ -500,7 +500,7 @@
    },
    "outputs": [],
    "source": [
-    "model.load_state_dict(torch.load(checkpoint, map_location=device))\n",
+    "model.load_state_dict(torch.load(checkpoint, map_location=device, weights_only=True))\n",
     "model.eval()\n",
     "results = []\n",
     "with torch.no_grad():\n",