diff --git a/modules/interpretability/class_lung_lesion.ipynb b/modules/interpretability/class_lung_lesion.ipynb
index 4b5db35e02..4aaf5d69ce 100644
--- a/modules/interpretability/class_lung_lesion.ipynb
+++ b/modules/interpretability/class_lung_lesion.ipynb
@@ -282,13 +282,16 @@
     "    ]\n",
     ")\n",
     "\n",
+    "persistent_cache = os.path.join(root_dir, \"persistent_cache\")\n",
     "train_ds = monai.data.PersistentDataset(\n",
-    "    data=train_files, transform=train_transforms\n",
+    "    data=train_files, transform=train_transforms, cache_dir=persistent_cache\n",
     ")\n",
     "train_loader = monai.data.DataLoader(\n",
     "    train_ds, batch_size=2, shuffle=True, num_workers=2, pin_memory=True\n",
     ")\n",
-    "val_ds = monai.data.PersistentDataset(data=val_files, transform=val_transforms)\n",
+    "val_ds = monai.data.PersistentDataset(\n",
+    "    data=val_files, transform=val_transforms, cache_dir=persistent_cache\n",
+    ")\n",
     "val_loader = monai.data.DataLoader(\n",
     "    val_ds, batch_size=2, num_workers=2, pin_memory=True\n",
     ")"
@@ -759,7 +762,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.7.10"
   }
  },
  "nbformat": 4,