fix for upgrade procedure