Fix build script
Fix build