diff --git a/pytools/py_codegen.py b/pytools/py_codegen.py
index ed689e15ec8326c880b992af502bf592eec5b8da..66ab67b0f42f712ad03d7efbbc869e29671b15fc 100644
--- a/pytools/py_codegen.py
+++ b/pytools/py_codegen.py
@@ -1,4 +1,4 @@
-from __future__ import division
+from __future__ import division, with_statement
 
 __copyright__ = "Copyright (C) 2009-2013 Andreas Kloeckner"