Make the build script python2 compatible