From 4b141b9fb59dbe354e49c5f3b37714f33b2f2cf9 Mon Sep 17 00:00:00 2001
From: Edward Tran <giaphutran012@gmail.com>
Date: Mon, 25 May 2026 13:20:07 +0700
Subject: [PATCH 1/2] Fix last table column resizing

---
 frontend/components/table/DatasetTable.tsx | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/frontend/components/table/DatasetTable.tsx b/frontend/components/table/DatasetTable.tsx
index 5644cff..4c1b1bc 100644
--- a/frontend/components/table/DatasetTable.tsx
+++ b/frontend/components/table/DatasetTable.tsx
@@ -21,6 +21,7 @@ const DEFAULT_COL_WIDTH = 180;
 const MIN_COL_WIDTH = 80;
 const ROW_HEIGHT = 34;
 const GHOST_ROW_COUNT = 50;
+const LAST_COLUMN_RESIZE_GUTTER = 160;
 
 const columnHelper = createColumnHelper<DatasetRow>();
 
@@ -68,6 +69,7 @@ export function DatasetTable({
   selection: Selection;
 }) {
   const tableContainerRef = useRef<HTMLDivElement>(null);
+  const previousResizingColumnIdRef = useRef<string | false>(false);
   const [containerHeight, setContainerHeight] = useState(600);
 
   useEffect(() => {
@@ -112,8 +114,16 @@ export function DatasetTable({
   const isBuilding = dataset.status === "building";
   const displayCount = isBuilding ? Math.max(tableRows.length, GHOST_ROW_COUNT) : tableRows.length;
   const totalWidth = columnWidths.reduce((sum, w) => sum + w, 0);
+  const tableContentWidth = totalWidth + LAST_COLUMN_RESIZE_GUTTER;
   const resizingColumnId = table.getState().columnSizingInfo.isResizingColumn;
 
+  useEffect(() => {
+    if (previousResizingColumnIdRef.current && !resizingColumnId) {
+      persistWidths();
+    }
+    previousResizingColumnIdRef.current = resizingColumnId;
+  }, [resizingColumnId, persistWidths]);
+
   const toggleRow = useCallback(
     (id: string, shiftKey: boolean) => {
       if (shiftKey) selection.shiftToggle(id);
@@ -143,7 +153,7 @@ export function DatasetTable({
         if (resizingColumnId) persistWidths();
       }}
     >
-      <div style={{ minWidth: totalWidth }}>
+      <div style={{ minWidth: tableContentWidth }}>
         <TableHeader
           headers={headers}
           columns={dataset.columns}

From d1b16d63257055fda76bf78982d2dcc13f95da12 Mon Sep 17 00:00:00 2001
From: Edward Tran <giaphutran012@gmail.com>
Date: Mon, 25 May 2026 17:03:34 +0700
Subject: [PATCH 2/2] Remove redundant resize mouseup handler

---
 frontend/components/table/DatasetTable.tsx | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/frontend/components/table/DatasetTable.tsx b/frontend/components/table/DatasetTable.tsx
index 4c1b1bc..18792da 100644
--- a/frontend/components/table/DatasetTable.tsx
+++ b/frontend/components/table/DatasetTable.tsx
@@ -149,9 +149,6 @@ export function DatasetTable({
       ref={tableContainerRef}
       className="flex-1 overflow-auto relative"
       style={{ fontSize: "13px" }}
-      onMouseUp={() => {
-        if (resizingColumnId) persistWidths();
-      }}
     >
       <div style={{ minWidth: tableContentWidth }}>
         <TableHeader