diff --git a/src/frontend/src/components/tableComponent/index.tsx b/src/frontend/src/components/tableComponent/index.tsx
index 507fe4674..cc43a4a47 100644
--- a/src/frontend/src/components/tableComponent/index.tsx
+++ b/src/frontend/src/components/tableComponent/index.tsx
@@ -139,10 +139,8 @@ const TableComponent = forwardRef<
           onGridReady={onGridReady}
           onColumnMoved={onColumnMoved}
           onStateUpdated={(e) => {
-            if (
-              e.sources.includes("columnVisibility") ||
-              e.sources.includes("columnOrder")
-            ) {
+            console.log(e);
+            if (e.sources.some((source) => source.includes("column"))) {
               setColumnStateChange(true);
             }
           }}