And apparently I forgot that import. It's:
try:
    from itertools import izip as zip
except ImportError:
    pass