diff --git a/tests/test_stocproc.py b/tests/test_stocproc.py
index 10517a3..61ebf34 100644
--- a/tests/test_stocproc.py
+++ b/tests/test_stocproc.py
@@ -26,9 +26,10 @@ from scipy.interpolate import InterpolatedUnivariateSpline
 from scipy.integrate import quad
 try:
     import matplotlib.pyplot as plt
+    from mpl_toolkits.axes_grid1 import make_axes_locatable
 except ImportError:
     print("matplotlib not found -> any plotting will crash")
-from mpl_toolkits.axes_grid1 import make_axes_locatable
+
 import functools
 import time