@@ -1,8 +1,5 @@
import logging
-try:
- from StringIO import StringIO
-except ImportError:
- from io import StringIO
+from io import StringIO
import sys
from mock import Mock