aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--tests/plugin_tests.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/plugin_tests.py b/tests/plugin_tests.py
index c4b540fa..31ffae40 100644
--- a/tests/plugin_tests.py
+++ b/tests/plugin_tests.py
@@ -4,9 +4,9 @@ import tempfile
# PYCOMPAT
import six
-if six.PY2:
+try:
from StringIO import StringIO
-else:
+except:
from io import StringIO
from sos.plugins import Plugin, regex_findall, sos_relative_path, mangle_command