HighLighting in CodeFolding

Hi folks,

I'm using the CustomFoldingBuilder to aim the code folding in my custom language. It works, but when the text is collapse the highlighting of it change to Comment Colour.

Is there any way to preserve the orginal colours of the tokens?

Thanks