Slight fix to update.js