diff --git a/generation/2d_autoencoderkl/2d_autoencoderkl_tutorial.ipynb b/generation/2d_autoencoderkl/2d_autoencoderkl_tutorial.ipynb
index cf47a232a..94c719f81 100644
--- a/generation/2d_autoencoderkl/2d_autoencoderkl_tutorial.ipynb
+++ b/generation/2d_autoencoderkl/2d_autoencoderkl_tutorial.ipynb
@@ -46,7 +46,7 @@
    "id": "2caa73e1",
    "metadata": {},
    "source": [
-    "## Set up environment"
+    "## Setup environment"
    ]
   },
   {
@@ -79,6 +79,7 @@
     "import os\n",
     "import shutil\n",
     "import time\n",
+    "import tempfile\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import torch\n",
diff --git a/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb b/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
index 0187bb9df..6d15d6e4e 100644
--- a/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
+++ b/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
@@ -74,6 +74,7 @@
    "source": [
     "import os\n",
     "import shutil\n",
+    "import tempfile\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import torch\n",
@@ -117,8 +118,7 @@
    ],
    "source": [
     "directory = os.environ.get(\"MONAI_DATA_DIRECTORY\")\n",
-    "root_dir = \"/tmp/tmp0_29f8jr\"\n",
-    "# root_dir = tempfile.mkdtemp() if directory is None else directory\n",
+    "root_dir = tempfile.mkdtemp() if directory is None else directory\n",
     "print(root_dir)"
    ]
   },