diff --git a/04_pytorch_custom_datasets.ipynb b/04_pytorch_custom_datasets.ipynb
index 99f00d04..51b5c1d2 100644
--- a/04_pytorch_custom_datasets.ipynb
+++ b/04_pytorch_custom_datasets.ipynb
@@ -1544,7 +1544,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 32,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -1586,7 +1586,7 @@
     "            title = f\"class: {classes[targ_label]}\"\n",
     "            if display_shape:\n",
     "                title = title + f\"\\nshape: {targ_image_adjust.shape}\"\n",
-    "        plt.title(title)"
+    "            plt.title(title)"
    ]
   },
   {