Fix broken import in Python 3.x