Make the import consistent between Python versions