👷♂️ Patch fixes
- Change prototype of element to the custom element when upgraded to keep the reference - By @capricorn86 in task #2331
- The previous solution replaced the element with the custom element and copied it's properties, but this made any reference to the original element invalid