diff --git a/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb b/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
index 97b637d24..c18e7f59d 100644
--- a/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
+++ b/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
@@ -178,6 +178,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "channel = 0  # 0 = Flair\n",
+    "assert channel in [0, 1, 2, 3], \"Choose a valid channel\"\n",
     "all_transforms = [\n",
     "    transforms.LoadImaged(keys=[\"image\"]),\n",
     "    transforms.EnsureChannelFirstd(keys=[\"image\"]),\n",
@@ -207,8 +209,6 @@
     }
    ],
    "source": [
-    "channel = 0  # 0 = Flair\n",
-    "assert channel in [0, 1, 2, 3], \"Choose a valid channel\"\n",
     "# Training\n",
     "train_ds = DecathlonDataset(\n",
     "    root_dir=root_dir,\n",
@@ -785,7 +785,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.0"
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,