diff --git a/components/lib/treetable/style/TreeTableStyle.js b/components/lib/treetable/style/TreeTableStyle.js
index 681f2f460..e6e66894c 100644
--- a/components/lib/treetable/style/TreeTableStyle.js
+++ b/components/lib/treetable/style/TreeTableStyle.js
@@ -456,7 +456,7 @@ const classes = {
         {
             'p-treetable-sortable-column': instance.columnProp('sortable'),
             'p-treetable-resizable-column': props.resizableColumns,
-            'p-treetable-column-sorted': instance.isColumnSorted(),
+            'p-treetable-column-sorted': context?.sorted,
             'p-treetable-frozen-column': instance.columnProp('frozen')
         }
     ],