- Fixed upgrade script