Little modification to the build script