Update for dist build process