Revising AI created corrections

When I use the AI feature in Lightroom Mobile, I select an area for it to revise, and it does its job by fixing that area. However, I wish I could modify the edited area after the AI has finished its work by using an erase or draw tool. 

For instance, in a rock garden edit, I might remove weeds or loose leaves, but the AI could select part of the model's foot in the process. While the rocks look perfect, the tip of her foot does not. If I like what the AI has generated and don’t want to lose that, I can’t fix the small flaw without losing the whole revision. I would like the ability to modify only the area I don’t want changed while keeping the parts I do like intact after AI generation.