diff --git a/scratch-vhdl-vscode/package.json b/scratch-vhdl-vscode/package.json
index 324d2aa..8b4bbda 100644
--- a/scratch-vhdl-vscode/package.json
+++ b/scratch-vhdl-vscode/package.json
@@ -8,7 +8,7 @@
   "publisher": "JosephAbbey",
   "displayName": "Scratch VHDL",
   "description": "",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "license": "MIT",
   "engines": {
     "vscode": "^1.75.0"
@@ -131,4 +131,4 @@
     "@vscode/webview-ui-toolkit": "^1.2.2",
     "blockly": "^10.1.3"
   }
-}
+}
\ No newline at end of file
diff --git a/scratch-vhdl-vscode/src/webview/TableEditor.ts b/scratch-vhdl-vscode/src/webview/TableEditor.ts
index 2a3e265..66c06d9 100644
--- a/scratch-vhdl-vscode/src/webview/TableEditor.ts
+++ b/scratch-vhdl-vscode/src/webview/TableEditor.ts
@@ -85,7 +85,7 @@ export default function TableEditor<
       const tr_td_input = document.createElement('input');
       tr_td.appendChild(tr_td_input);
       tr.appendChild(tr_td);
-      for (let i = 0; i < Object.values(rows)[0].length; i++) {
+      for (let i = 0; i < cols.length - 1; i++) {
         const td = document.createElement('td');
         const td_input = document.createElement('input');
         td.appendChild(td_input);