pythonthings
diff --git a/‎bpython/test/test_curtsies_repl.py‎
Lines changed: 6 additions & 2 deletions b/‎bpython/test/test_curtsies_repl.py‎
Lines changed: 6 additions & 2 deletions
@@ -5,7 +5,6 @@
 from contextlib import contextmanager
 from mock import Mock, patch, MagicMock
 import os
-from StringIO import StringIO
 import sys
 import tempfile
 
@@ -14,6 +13,11 @@
 except ImportError:
     import unittest
 
+try:
+    from io import BytesIO
+except ImportError:
+    from StringIO import StringIO as BytesIO
+
 from bpython.curtsiesfrontend import repl as curtsiesrepl
 from bpython.curtsiesfrontend import interpreter
 from bpython import autocomplete
@@ -201,7 +205,7 @@ def test_list_win_not_visible_and_match_selected_if_one_option(self):
 
 @contextmanager # from http://stackoverflow.com/a/17981937/398212 - thanks @rkennedy
 def captured_output():
-    new_out, new_err = StringIO(), StringIO()
+    new_out, new_err = BytesIO(), BytesIO()
     old_out, old_err = sys.stdout, sys.stderr
     try:
         sys.stdout, sys.stderr = new_out, new_err