Fix import on Python 3.