diff --git a/_episodes/03-matplotlib.md b/_episodes/03-matplotlib.md
index b14c5ca63b..f837a50cdf 100644
--- a/_episodes/03-matplotlib.md
+++ b/_episodes/03-matplotlib.md
@@ -360,7 +360,7 @@ which is the default for plotting matrix-type data (because this is where `[0:0]
 We can also add a colour bar to help describe the figure, with a little more code:
 
 ~~~
-import mpl_toolkits
+import mpl_toolkits.axes_grid1
 
 matplotlib.pyplot.figure()
 ax = matplotlib.pyplot.gca()