Fixed the UMD wrapper