diff --git a/python_modules/codegen.py b/python_modules/codegen.py
index af6636b..03c67e6 100644
--- a/python_modules/codegen.py
+++ b/python_modules/codegen.py
@@ -1,3 +1,4 @@
+from __future__ import with_statement
 from cStringIO import StringIO
 
 def camel_to_underscores(s, upper = False):
diff --git a/python_modules/demarshal.py b/python_modules/demarshal.py
index c588f98..cbe3599 100644
--- a/python_modules/demarshal.py
+++ b/python_modules/demarshal.py
@@ -1,3 +1,4 @@
+from __future__ import with_statement
 import ptypes
 import codegen
 
diff --git a/python_modules/marshal.py b/python_modules/marshal.py
index 6b894c1..9ee1466 100644
--- a/python_modules/marshal.py
+++ b/python_modules/marshal.py
@@ -1,3 +1,4 @@
+from __future__ import with_statement
 import ptypes
 import codegen