Tweak the build script