fixing upgrade issue found not working properly