diff --git a/website/demos/RowGrouping.tsx b/website/demos/RowGrouping.tsx
index 18bbafdff0..b31e007d53 100644
--- a/website/demos/RowGrouping.tsx
+++ b/website/demos/RowGrouping.tsx
@@ -203,7 +203,10 @@ export default function RowGrouping({ direction }: Props) {
       </label>
 
       <TreeDataGrid
-        columns={columns.map((c) => ({ ...c, frozen: areGroupColumnsFrozen }))}
+        columns={columns.map((c) => ({
+          ...c,
+          ...(selectedOptions.includes(c.key) ? { frozen: areGroupColumnsFrozen } : {})
+        }))}
         rows={rows}
         rowKeyGetter={rowKeyGetter}
         selectedRows={selectedRows}