Update build script

Change-Id: Ie51f90569145454eb52776a5036b82650b4fbc3c