Update AI Settings

I am using the AI remove tool for 1 blemish on a photo. I update it (re generate) right before saving and I am still sometimes unable to save the photo with the AI edit - I am getting a weird square around the area I am trying to fix. I have been using this for months and never had this issue - Im not doing anything differently, it seems like it must be a bug.