1 files changed, 5 insertions, 4 deletions
diff --git a/tests/unit/test_processutils.py b/tests/unit/test_processutils.py
index e00a66e..fbe9f34 100644
--- a/tests/unit/test_processutils.py
+++ b/tests/unit/test_processutils.py
@@ -19,9 +19,10 @@ from __future__ import print_function
 
 import fixtures
 import os
-import StringIO
 import tempfile
 
+from six import StringIO
+
 from openstack.common import processutils
 from tests import utils
 
@@ -213,7 +214,7 @@ class FakeSshChannel(object):
         return self.rc
 
 
-class FakeSshStream(StringIO.StringIO):
+class FakeSshStream(StringIO):
     def setup_channel(self, rc):
         self.channel = FakeSshChannel(rc)
 
@@ -225,9 +226,9 @@ class FakeSshConnection(object):
     def exec_command(self, cmd):
         stdout = FakeSshStream('stdout')
         stdout.setup_channel(self.rc)
-        return (StringIO.StringIO(),
+        return (StringIO(),
                 stdout,
-                StringIO.StringIO('stderr'))
+                StringIO('stderr'))
 
 
 class SshExecuteTestCase(utils.BaseTestCase):