diff --git a/docs/sphinxext/ipython_directive.py b/docs/sphinxext/ipython_directive.py
index 33ba664..d18a0be 100644
--- a/docs/sphinxext/ipython_directive.py
+++ b/docs/sphinxext/ipython_directive.py
@@ -62,6 +62,7 @@ import os
 import re
 import sys
 import tempfile
+import ast
 
 # To keep compatibility with various python versions
 try: