diff --git a/notebooks/shared/ClassDiagram.ipynb b/notebooks/shared/ClassDiagram.ipynb
index a981a0538..3fba8034c 100644
--- a/notebooks/shared/ClassDiagram.ipynb
+++ b/notebooks/shared/ClassDiagram.ipynb
@@ -45,7 +45,7 @@
    },
    "outputs": [],
    "source": [
-    "import bookutils"
+    "import bookutils.setup"
    ]
   },
   {
diff --git a/notebooks/shared/ExpectError.ipynb b/notebooks/shared/ExpectError.ipynb
index a94421977..79f4e791c 100644
--- a/notebooks/shared/ExpectError.ipynb
+++ b/notebooks/shared/ExpectError.ipynb
@@ -130,7 +130,7 @@
    },
    "outputs": [],
    "source": [
-    "import bookutils"
+    "import bookutils.setup"
    ]
   },
   {
diff --git a/notebooks/shared/Timeout.ipynb b/notebooks/shared/Timeout.ipynb
index b2c557525..ca929c00f 100644
--- a/notebooks/shared/Timeout.ipynb
+++ b/notebooks/shared/Timeout.ipynb
@@ -102,7 +102,7 @@
    },
    "outputs": [],
    "source": [
-    "import bookutils"
+    "import bookutils.setup"
    ]
   },
   {
diff --git a/notebooks/shared/Timer.ipynb b/notebooks/shared/Timer.ipynb
index 2219f9d73..cef67fc67 100644
--- a/notebooks/shared/Timer.ipynb
+++ b/notebooks/shared/Timer.ipynb
@@ -85,7 +85,7 @@
    },
    "outputs": [],
    "source": [
-    "import bookutils"
+    "import bookutils.setup"
    ]
   },
   {