1 files changed, 4 insertions, 6 deletions
diff --git a/test/test_writer.py b/test/test_writer.py
index a3f3893..f8186d1 100644
--- a/test/test_writer.py
+++ b/test/test_writer.py
@@ -1,10 +1,6 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, print_function, unicode_literals
-
-try:
-    import unittest2 as unittest
-except ImportError:
-    import unittest
+import unittest
 import test
 import yamlish
 
@@ -191,10 +187,12 @@ test_data_list = [
   },
 ]
 
+
 class TestWriter(unittest.TestCase):  # IGNORE:C0111
     pass
 
-test.generate_testsuite(test_data_list, TestWriter, yamlish.dump, direction=test.OUTPUT)
+test.generate_testsuite(test_data_list, TestWriter, yamlish.dump,
+                        direction=test.OUTPUT)
 
 if __name__ == "__main__":
     unittest.main()