fix for upgrade