minor tweak to python_support

2016-12-06 23:09:55 -05:00 · 2016-12-06 23:09:55 -05:00 · 0fc44efbcc
commit 0fc44efbcc
parent 1b36ca00e5
1 changed files with 3 additions and 1 deletions
--- a/mongoengine/python_support.py
+++ b/mongoengine/python_support.py
@ -22,6 +22,8 @@ StringIO = six.BytesIO
 # Additionally for Py2, try to use the faster cStringIO, if available
 if not PY3:
    try:
-        from cStringIO import StringIO
+        import cStringIO
    except ImportError:
        pass
+    else:
+        StringIO = cStringIO.StringIO