Fix for the update
The actual update fix

Tested and it works