diff --git a/priv/python2/erlport/erlang.py b/priv/python2/erlport/erlang.py
index <HASH>..<HASH> 100644
--- a/priv/python2/erlport/erlang.py
+++ b/priv/python2/erlport/erlang.py
@@ -25,6 +25,10 @@
 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 # POSSIBILITY OF SUCH DAMAGE.
 
+from __future__ import with_statement
+
+__author__ = "Dmitry Vasiliev <dima@hlabs.org>"
+
 import sys
 from sys import exc_info
 from traceback import extract_tb
diff --git a/priv/python2/erlport/erlproto.py b/priv/python2/erlport/erlproto.py
index <HASH>..<HASH> 100644
--- a/priv/python2/erlport/erlproto.py
+++ b/priv/python2/erlport/erlproto.py
@@ -27,6 +27,8 @@
 
 """Erlang port protocol."""
 
+from __future__ import with_statement
+
 __author__ = "Dmitry Vasiliev <dima@hlabs.org>"
 
 import os
Add 'with' statement declaration for Python <I>