diff --git a/apps/client/src/widgets/view_widgets/table_view/data.ts b/apps/client/src/widgets/view_widgets/table_view/data.ts
index d798b345e..d89a449f6 100644
--- a/apps/client/src/widgets/view_widgets/table_view/data.ts
+++ b/apps/client/src/widgets/view_widgets/table_view/data.ts
@@ -69,7 +69,7 @@ export function buildColumnDefinitions(info: PromotedAttributeInformation[]) {
         columnDefs.push({
             field: `labels.${name}`,
             title: title ?? name,
-            editable: true
+            editor: "input"
             // cellDataType: mapDataType(type),
         });
     }
diff --git a/apps/client/src/widgets/view_widgets/table_view/index.ts b/apps/client/src/widgets/view_widgets/table_view/index.ts
index 53bc5da8a..f06592175 100644
--- a/apps/client/src/widgets/view_widgets/table_view/index.ts
+++ b/apps/client/src/widgets/view_widgets/table_view/index.ts
@@ -117,30 +117,16 @@ export default class TableView extends ViewMode<StateInfo> {
             const field = cell.getField();
             const newValue = cell.getValue();
 
-            console.log("Cell edited", field, newValue);
             if (field === "title") {
                 server.put(`notes/${noteId}/title`, { title: newValue });
+                return;
+            }
+
+            if (field.startsWith("labels.")) {
+                const labelName = field.split(".", 2)[1];
+                setLabel(noteId, labelName, newValue);
             }
         });
-
-        // return {
-        //     onCellValueChanged(event) {
-        //         if (event.type !== "cellValueChanged") {
-        //             return;
-        //         }
-
-        //         const noteId = event.data.noteId;
-        //         const name = event.colDef.field;
-        //         if (!name) {
-        //             return;
-        //         }
-
-        //         if (name.startsWith("labels.")) {
-        //             const labelName = name.split(".", 2)[1];
-        //             setLabel(noteId, labelName, newValue);
-        //         }
-        //     }
-        // }
     }
 
     private setupDragging() {