Update upgrade.js