diff --git a/pytti_5_beta.ipynb b/pytti_5_beta.ipynb
index 9587c47..dfee161 100644
--- a/pytti_5_beta.ipynb
+++ b/pytti_5_beta.ipynb
@@ -647,7 +647,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from pytti.pytti_cli_w_clearml import _main, TB_LOGDIR \n",
+        "from pytti.workhorse import _main, TB_LOGDIR \n",
         "\n",
         "%load_ext tensorboard"
       ]