--- a/contrib/python-zstandard/tests/test_buffer_util.py	Sun Apr 08 01:08:43 2018 +0200
+++ b/contrib/python-zstandard/tests/test_buffer_util.py	Mon Apr 09 10:13:29 2018 -0700
@@ -1,11 +1,7 @@
 import struct
+import unittest
 
-try:
-    import unittest2 as unittest
-except ImportError:
-    import unittest
-
-import zstd
+import zstandard as zstd
 
 ss = struct.Struct('=QQ')