diff --git a/test/test_prefix.py b/test/test_prefix.py
index 369e6c8..0c79958 100644
--- a/test/test_prefix.py
+++ b/test/test_prefix.py
@@ -1,4 +1,9 @@
-from itertools import zip_longest
+try:
+    from itertools import zip_longest
+except ImportError:
+    # Python 2
+    from itertools import izip_longest as zip_longest
+
 from codecs import BOM_UTF8
 
 import pytest