I think it is a different issue. Sometimes the icon just does not work, but you can see it. You click “proofread” and afterwards you again see the “AI” button. I noticed this on different devices.
Also, sometimes the modal showing the changes is just blank even though there are changes. But I wasn’t able to notice a pattern, so I have no steps to reproduce.