Fixing the build script