Fixed more issues with the update process