slight fix to update script