Retaining units after changing strata and probabilities