How can you add a sorting class to the headers and not just the columns?

Title says it all. I get aria-labels and controls but no actual class on th elements. I do see a span that changes class based on triangles, but these aren't really styleable for my needs. I need to add a background color to the th that is sorting.