fixing upgrade