Update build to work