Update build script to fix compiling problem