Fix highlighting toggle so it works both within the application preview and the exported file.