diff --git a/examples/Demo of CMHE on Synthetic Data.ipynb b/examples/Demo of CMHE on Synthetic Data.ipynb
index cda45fa..29eaa56 100644
--- a/examples/Demo of CMHE on Synthetic Data.ipynb	
+++ b/examples/Demo of CMHE on Synthetic Data.ipynb	
@@ -88,7 +88,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 36,
    "id": "a99d9a37",
    "metadata": {},
    "outputs": [],
@@ -100,7 +100,7 @@
     "sys.path.append('../')\n",
     "\n",
     "from auton_survival.datasets import load_dataset\n",
-    "from example_utils import * "
+    "from cmhe_demo_utils import * "
    ]
   },
   {