diff --git a/modules/mmcif/test/test_convert.py b/modules/mmcif/test/test_convert.py
index fb1b6143dd..d7ccbea614 100644
--- a/modules/mmcif/test/test_convert.py
+++ b/modules/mmcif/test/test_convert.py
@@ -5,12 +5,8 @@
 import ihm
 import ihm.reader
 import os
-import sys
 from test_restraints import MockGaussianEMRestraint
-if sys.version_info[0] >= 3:
-    from io import StringIO
-else:
-    from io import BytesIO as StringIO
+from io import StringIO
 
 
 class Tests(IMP.test.TestCase):
diff --git a/modules/mmcif/test/test_dumper.py b/modules/mmcif/test/test_dumper.py
index 9cbc7fc48f..188ce1d2f5 100644
--- a/modules/mmcif/test/test_dumper.py
+++ b/modules/mmcif/test/test_dumper.py
@@ -4,13 +4,8 @@
 import ihm.dataset
 import ihm.format
 import ihm.dumper
-import io
-import sys
 import os
-if sys.version_info[0] >= 3:
-    from io import StringIO
-else:
-    from io import BytesIO as StringIO
+from io import StringIO
 
 class MockGaussianEMRestraint(IMP.Restraint):