Quick fix to upgrade checker