diff --git a/test/test_algorithm.py b/test/test_algorithm.py
index 7f0f9f4c71b8ab4b73f79d4fd7afbe4537677201..06a9258032920bee78ee4d56c650c0cbb830eb83 100644
--- a/test/test_algorithm.py
+++ b/test/test_algorithm.py
@@ -1,5 +1,7 @@
 #! /usr/bin/env python
 
+from __future__ import division, with_statement
+
 __copyright__ = "Copyright (C) 2013 Andreas Kloeckner"
 
 __license__ = """
diff --git a/test/test_array.py b/test/test_array.py
index b70deb509d9c8e50d4fe09b63d11a6748eef6e75..49c9d44bd3d7d317e720dcf9e712e78d83c0cc23 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -1,5 +1,5 @@
 #! /usr/bin/env python
-from __future__ import division
+from __future__ import division, with_statement
 
 __copyright__ = "Copyright (C) 2009 Andreas Kloeckner"