Fixes minor bug in upgrade