Thanks for the screenshots - that makes it clear. Highlighting works while the block is streaming, then drops once generation finishes.
This is a known CLI rendering issue on our side. The code content is fine; only the colors in completed scrollback can be lost, and there’s no reliable workaround on your end right now.
I’ve attached your report to the tracking item so it covers live sessions as well as resumed sessions. We’ll follow up when a fix ships.