1 files changed, 5 insertions, 0 deletions
diff --git a/mako/compat.py b/mako/compat.py
index 3a4742d..5e9c201 100644
--- a/mako/compat.py
+++ b/mako/compat.py
@@ -18,6 +18,8 @@ if py3k:
     binary_type = bytes
     text_type = str
 
+    from io import BytesIO as byte_buffer
+
     def u(s):
         return s
 
@@ -30,6 +32,9 @@ else:
         from cStringIO import StringIO
     except:
         from StringIO import StringIO
+
+    byte_buffer = StringIO
+
     from urllib import quote_plus, unquote_plus
     from htmlentitydefs import codepoint2name, name2codepoint
     string_types = basestring,