diff --git a/Makefile b/Makefile
index 34b67fc458..e6cc00207e 100644
--- a/Makefile
+++ b/Makefile
@@ -31,3 +31,6 @@ test: clean
 docs: clean
 	pip install --quiet -r requirements/docs.txt
 	python -m sphinx -b html -W --keep-going docs/source docs/build
+
+update:
+	git submodule update --init --recursive --remote
diff --git a/_notebooks b/_notebooks
index 0c32582910..290fb466de 160000
--- a/_notebooks
+++ b/_notebooks
@@ -1 +1 @@
-Subproject commit 0c325829101d5a6ebf32ed99bbf5b09badf04a59
+Subproject commit 290fb466de1fcc2ac6025f74b56906592911e856
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 2c858bc21a..a111a218ae 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -30,8 +30,8 @@ sys.path.append(os.path.join(PATH_RAW_NB, ".actions"))
 _SHOULD_COPY_NOTEBOOKS = True
 
 try:
-    from helpers import HelperCLI
-except Exception:
+    from assistant import AssistantCLI
+except ImportError:
     _SHOULD_COPY_NOTEBOOKS = False
     warnings.warn("To build the code, please run: `git submodule update --init --recursive`", stacklevel=2)
 
@@ -46,7 +46,9 @@ spec.loader.exec_module(about)
 
 # -- Project documents -------------------------------------------------------
 if _SHOULD_COPY_NOTEBOOKS:
-    HelperCLI.copy_notebooks(PATH_RAW_NB, PATH_HERE, "notebooks", patterns=[".", "course_UvA-DL", "lightning_examples"])
+    AssistantCLI.copy_notebooks(
+        PATH_RAW_NB, PATH_HERE, "notebooks", patterns=[".", "course_UvA-DL", "lightning_examples"]
+    )
 
 
 def _transform_changelog(path_in: str, path_out: str) -> None: