Restore auto-update functionality