minor change in upgrade routine