Fix an importerror when run under python3.