Fix import statement in Python 3