diff --git a/apps/server/src/services/llm/tools.ts b/apps/server/src/services/llm/tools.ts
index d4ab67e821..bbdbf6ec3a 100644
--- a/apps/server/src/services/llm/tools.ts
+++ b/apps/server/src/services/llm/tools.ts
@@ -107,7 +107,7 @@ export const updateNoteContent = tool({
         if (note.isProtected) {
             return { error: "Note is protected and cannot be modified" };
         }
-        if (note.type !== "text" && note.type !== "code") {
+        if (!note.hasStringContent()) {
             return { error: `Cannot update content for note type: ${note.type}` };
         }