diff --git a/docs/tutorials/contribute_non_geo_dataset.ipynb b/docs/tutorials/contribute_non_geo_dataset.ipynb
index 4a702222daf..2ea23365a41 100644
--- a/docs/tutorials/contribute_non_geo_dataset.ipynb
+++ b/docs/tutorials/contribute_non_geo_dataset.ipynb
@@ -219,13 +219,15 @@
    "source": [
     "import os\n",
     "import shutil\n",
+    "import tempfile\n",
     "\n",
     "import numpy as np\n",
     "from PIL import Image\n",
     "\n",
     "# Define the root directory and subdirectories\n",
-    "root_dir = 'my_new_dataset'\n",
-    "sub_dirs = ['sub_dir_1', 'sub_dir_2', 'sub_dir_3']\n",
+    "# Normally this would be the current directory (tests/data/my_new_dataset)\n",
+    "root_dir = os.path.join(tempfile.gettempdir(), 'my_new_dataset')\n",
+    "sub_dirs = ['image', 'target']\n",
     "splits = ['train', 'val', 'test']\n",
     "\n",
     "image_file_names = ['sample_1.png', 'sample_2.png', 'sample_3.png']\n",