diff --git a/test/test_pytools.py b/test/test_pytools.py
index 7cecf68315d047a840ed04314b670101fb3abeb9..0be7356697e68711f75d40f1994fa8f2ca1563a2 100644
--- a/test/test_pytools.py
+++ b/test/test_pytools.py
@@ -1,4 +1,4 @@
-from __future__ import division
+from __future__ import division, with_statement
 
 import pytest
 import sys  # noqa