trying to fix upgrade unit levels