Fix UMD checks