Fix issue in build script