Fix UMD issue