Undo button reverts AI edits (and can't be redone)

In the editing interface, the undo button in the “Color” section also undoes the paid AI edits. Generally that would be ok, but then that change can’t be reverted with the redo button. I had to repeat the edit and spend again credits.

Note: I can’t change the type of this post, so it’s listed as a feature request, but it’s really a bug report.