fix upgrade again