diff --git a/tuskarclient/tests/common/test_formatting.py b/tuskarclient/tests/common/test_formatting.py
index d315c6e..6d26626 100644
--- a/tuskarclient/tests/common/test_formatting.py
+++ b/tuskarclient/tests/common/test_formatting.py
@@ -10,8 +10,8 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
-import io
 import mock
+import six
 
 import tuskarclient.common.formatting as fmt
 import tuskarclient.tests.utils as tutils
@@ -21,7 +21,7 @@ class PrintTest(tutils.TestCase):
 
     def setUp(self):
         super(PrintTest, self).setUp()
-        self.outfile = io.StringIO()
+        self.outfile = six.StringIO()
 
     def test_print_dict(self):
         dict_ = {'k': 'v', 'key': 'value'}
diff --git a/tuskarclient/tests/v1/test_overcloud_roles_shell.py b/tuskarclient/tests/v1/test_overcloud_roles_shell.py
index 3792d6d..538cb34 100644
--- a/tuskarclient/tests/v1/test_overcloud_roles_shell.py
+++ b/tuskarclient/tests/v1/test_overcloud_roles_shell.py
@@ -10,8 +10,8 @@
 # License for the specific language governing permissions and limitations
 # under the License.
 
-import io
 import mock
+import six
 
 import tuskarclient.tests.utils as tutils
 from tuskarclient.v1 import overcloud_roles_shell
@@ -36,7 +36,7 @@ class RacksShellTest(tutils.TestCase):
 
     def setUp(self):
 
-        self.outfile = io.StringIO()
+        self.outfile = six.StringIO()
         self.tuskar = mock.MagicMock()
         self.shell = overcloud_roles_shell
         super(RacksShellTest, self).setUp()
diff --git a/tuskarclient/tests/v1/test_overclouds_shell.py b/tuskarclient/tests/v1/test_overclouds_shell.py
index e6c7aa0..976dba0 100644
--- a/tuskarclient/tests/v1/test_overclouds_shell.py
+++ b/tuskarclient/tests/v1/test_overclouds_shell.py
@@ -10,8 +10,8 @@
 # License for the specific language governing permissions and limitations
 # under the License.
 
-import io
 import mock
+import six
 
 import tuskarclient.tests.utils as tutils
 from tuskarclient.v1 import overclouds_shell
@@ -36,7 +36,7 @@ class RacksShellTest(tutils.TestCase):
 
     def setUp(self):
 
-        self.outfile = io.StringIO()
+        self.outfile = six.StringIO()
         self.tuskar = mock.MagicMock()
         self.shell = overclouds_shell
         super(RacksShellTest, self).setUp()