At the risk of pressing on a known pain point: I didn't mind so much losing the highlighting upon entering edit mode, but I did mind losing it forever, i.e. no

At the risk of pressing on a known pain point: I didn't mind so much losing the highlighting upon entering edit mode, but I did mind losing it forever, i.e. not having it even after leaving edit mode. Perhaps if there was a way to re-apply the syntax highlighting once the editable element loses focus, then losing syntax highlighting temporarily when actively editing wouldn't be as much of a problem?