UX - preserving context when returning from UI use cases

Steps to reproduce:

1. Go to host groups
2. Click on template
3. Update or cancel

Result:
Returned to templates, not host groups. It is a bit confusing, especially for new users.
Expected:
Just an idea, but maybe frontend could preserve context, from where user go to the use case, and return him to that original context on exit from the use case.