Feature - 3rd Party Audit Comments

When I am carrying out an audit of a third party. If I add comments to the audit section I am looking at it takes me back to /complianceAudits and I need to click back in to the audit to continue.

It would be much better if like the compliance analysis that it just took you back to the last part you were working on