Fix build
Update build task