Undo feature don't work after PIE (play in editor)

Back in 4.20, I was able to do some changes, play in editor, and after resuming play in editor I could use CTRL+Z to undo the changes I had done BEFORE playing in editor.

In 4.21, this doesn’t seem to be possible, when playing in editor the undo history resets. It was a GREAT feature that made iteration process fast and safe. Please, could you restore this functionality?