Fix a Python 3 import error.