diff --git a/doc/ext/notebook.py b/doc/ext/notebook.py
index b62401fa..43ede80c 100644
--- a/doc/ext/notebook.py
+++ b/doc/ext/notebook.py
@@ -1,3 +1,5 @@
+__all__ = ['python_to_notebook']
+
 import json
 import copy
 
diff --git a/doc/ext/plot2rst.py b/doc/ext/plot2rst.py
index e951729f..07b54470 100644
--- a/doc/ext/plot2rst.py
+++ b/doc/ext/plot2rst.py
@@ -80,7 +80,7 @@ from skimage import io
 from skimage import transform
 from skimage.util.dtype import dtype_range
 
-from notebook import save_ipython_notebook
+from notebook import python_to_notebook
 
 
 LITERALINCLUDE = """