Fix import path for Python3