Changing behaviour when clicking on internal links (editing mode)

Current situtation

I am always in editing mode with live preview but internal links behave poorly visually in this mode.
When clicking on an internal link, it goes to the section but it also highlights everything under the header, like the whole page, so every time I have to re-click somewhere to make the highlight disappears.
Also it puts the caret on the line, disabling the preview. It’s not great on header but it is particularly bad on callouts because they go full edit mode. And I love callouts.

What I would like

I would like to either highlight only the title (and with the accent color not a ugly yellow ) or even not highlight at all.
And I would like to simply move to the section but without moving the caret. Or maybe moving it a line above so it doesn’t mess up the preview.
I am opened to other ideas though.