diff --git a/fuelclient/tests/unit/v2/cli/test_node.py b/fuelclient/tests/unit/v2/cli/test_node.py
index f15d28e..b0eec12 100644
--- a/fuelclient/tests/unit/v2/cli/test_node.py
+++ b/fuelclient/tests/unit/v2/cli/test_node.py
@@ -14,13 +14,14 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
-import io
 import json
 
 import mock
 import six
 import yaml
 
+from six import StringIO
+
 from fuelclient import main as main_mod
 from fuelclient.tests.unit.v2.cli import test_engine
 from fuelclient.tests.utils import fake_node
@@ -119,7 +120,7 @@ node-4 ansible_host=10.20.0.5
 '''
 
         args = 'node ansible-inventory --env 1'
-        with mock.patch('sys.stdout', new=io.StringIO()) as mstdout:
+        with mock.patch('sys.stdout', new=StringIO()) as mstdout:
             rv = self.exec_command(args)
             actual_output = mstdout.getvalue()