diff --git a/doc/conf.py b/doc/conf.py
index 9cf154a1..801eb65b 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -20,7 +20,7 @@
     "sphinx.ext.napoleon",
     "nbsphinx",
     "sphinx_gallery.load_style",
-    "sphinx_toolbox",
+    "sphinx_toolbox.code",
 ]