Fix UMD build