diff --git a/test/refactor.py b/test/refactor.py
index d09c3d35..92923b09 100755
--- a/test/refactor.py
+++ b/test/refactor.py
@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+from __future__ import with_statement
 import sys
 import os
 import traceback