Update build to work with the modified UMD.