Federal prosecutors say they’ll drop Michael Avenatti’s remaining 31 charges if his sentence for last week’s guilty pleas accounts for the full scope of his crimes, heightening an already high-stakes September hearing.

The move is an offer to Senior U.S. District Judge James V. Selna to impose the sentence prosecutors’ request, and in exchange, rid himself of another trial for Avenatti. But it also puts pressure on Avenatti to drop the adversarial approach he vowed to take when pleading guilty last week, when he told Selna he’ll be arguing that the total amount of money of which he defrauded his clients is “drastically less” than the $9 million prosecutors stated.