Fixed bug in upgrade script.