Improve rollback command in build script