diff --git a/main.ipynb b/main.ipynb
index 3d2d095dc242ad1d41296300fac97bfd9ab77664..56f9685c6a43f3e76216707292125aa41a16b5a1 100644
--- a/main.ipynb
+++ b/main.ipynb
@@ -303,7 +303,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "NAME = 'Baseline_VAEP_short_512_10=warmup_begin'\n",
+    "NAME = 'Baseline_VAE_short_512_10=warmup_begin'\n",
     "log_dir = os.path.abspath(\n",
     "    \"logs/fit/{}_{}\".format(NAME, datetime.now().strftime(\"%Y%m%d-%H%M%S\"))\n",
     ")\n",