How does one handle undo action?

How do I shift the focus from the plugin over to the app after I’ve executed some actions on the document through the plugin? I want to avoid having to click on the document every time I have to undo. Currently the focus still stays on the plugin after I’ve clicked a button in it - hence “command + z” doesn’t work.

From ui.html:;
// Or
1 Like