diff --git a/IPython/nbconvert/transformers/extractfigure.py b/IPython/nbconvert/transformers/extractfigure.py
index 0df9814..d9d0fb3 100755
--- a/IPython/nbconvert/transformers/extractfigure.py
+++ b/IPython/nbconvert/transformers/extractfigure.py
@@ -13,6 +13,7 @@ notebook file.  The extracted figures are returned in the 'resources' dictionary
 # Imports
 #-----------------------------------------------------------------------------
 
+import sys
 from IPython.utils.traitlets import Dict, Unicode
 from .activatable import ActivatableTransformer