Fix build script

Fix build script