fixed upgrade issue