When accessing rule/element from "edit library" screen, return back to "edit library" screen

It's great that you can get from the "edit library" screen into a rule/element to compare revisions.  However, it is frustrating that when you close the rule/element you lose the context and get kicked out to the element/rules list rather than back to the edit library page.

How it currently works:

Edit Library > view rule > compare revision > *close* compare revision > *close* rule > rule list page 😞

How it should work:

Edit Library > view rule > compare revision > *close* compare revision > *close* rule > Edit Library 🙂

Please can you consider changing the work flow to remember the context in which the rule was accessed so you are returned to the appropriate place? - when looking at a lot of rules this would really make a difference!