diff --git a/scipy-notebook/test/test_matplotlib.py b/scipy-notebook/test/test_matplotlib.py
index 509c285e..34a63f79 100644
--- a/scipy-notebook/test/test_matplotlib.py
+++ b/scipy-notebook/test/test_matplotlib.py
@@ -3,6 +3,7 @@
 
 import logging
 
+import pytest
 import os
 
 LOGGER = logging.getLogger(__name__)