diff --git a/webview/src/stories/Table.stories.tsx b/webview/src/stories/Table.stories.tsx
index a9367631b3..b607d9bb26 100644
--- a/webview/src/stories/Table.stories.tsx
+++ b/webview/src/stories/Table.stories.tsx
@@ -366,7 +366,8 @@ WithMultipleBranches.args = {
   tableData: getTableState({
     ...survivalTableData,
     rows: [
-      ...survivalTableData.rows.map(row => ({
+      survivalTableData.rows[0],
+      ...rowsWithoutWorkspace.map(row => ({
         ...row,
         branch: branches[0],
         subRows: row.subRows?.map(subRow => ({