Fixing mistake
Bug fix for upgrade.js