Update build script

Change-Id: Ic6596c090161263cd83018a8d8a2ff873840c237