diff --git a/flask_peewee/tests/admin.py b/flask_peewee/tests/admin.py
index <HASH>..<HASH> 100644
--- a/flask_peewee/tests/admin.py
+++ b/flask_peewee/tests/admin.py
@@ -1,3 +1,5 @@
+from __future__ import with_statement
+
 import datetime
 
 from flask import request, session, url_for
diff --git a/flask_peewee/tests/auth.py b/flask_peewee/tests/auth.py
index <HASH>..<HASH> 100644
--- a/flask_peewee/tests/auth.py
+++ b/flask_peewee/tests/auth.py
@@ -1,3 +1,5 @@
+from __future__ import with_statement
+
 import datetime
 
 from flask import request, session, url_for, get_flashed_messages
diff --git a/flask_peewee/tests/rest.py b/flask_peewee/tests/rest.py
index <HASH>..<HASH> 100644
--- a/flask_peewee/tests/rest.py
+++ b/flask_peewee/tests/rest.py
@@ -1,3 +1,5 @@
+from __future__ import with_statement
+
 try:
     import simplejson as json
 except ImportError:
Python <I> fixes for the test suite, which uses with statements