diff --git a/demo.ipynb b/demo.ipynb
index b6be54f..5a784f9 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -207,7 +207,7 @@
     "    model=build_networks_from_config(config),\n",
     "    noise_scheduler=build_noise_scheduler_from_config(config),\n",
     "    normalizer=LinearNormalizer(),\n",
-    "    clf_cbf_controller=clf_cbf_controller,  # set as clf_cbf_controller to enable CLF-CBF traj refinement\n",
+    "    clf_cbf_controller=None,  # set as clf_cbf_controller to enable CLF-CBF traj refinement\n",
     "    config=config,\n",
     "    device=\"cuda\" if torch.cuda.is_available() else \"cpu\",\n",
     ")"