diff --git a/packages/client/src/handlers/mutations/messages/message-create.ts b/packages/client/src/handlers/mutations/messages/message-create.ts
index 45cb10c7..5bd18974 100644
--- a/packages/client/src/handlers/mutations/messages/message-create.ts
+++ b/packages/client/src/handlers/mutations/messages/message-create.ts
@@ -36,7 +36,7 @@ export class MessageCreateMutationHandler
     // check if there are nested nodes (files, pages, folders etc.)
     for (const block of Object.values(blocks)) {
       if (block.type === EditorNodeTypes.TempFile) {
-        const tempFileId = block.attrs?.id;
+        const tempFileId = block.id;
         if (!tempFileId) {
           throw new MutationError(
             MutationErrorCode.FileInvalid,
diff --git a/packages/ui/src/editor/extensions/temp-file.tsx b/packages/ui/src/editor/extensions/temp-file.tsx
index 3413ba2e..e8c8a0a2 100644
--- a/packages/ui/src/editor/extensions/temp-file.tsx
+++ b/packages/ui/src/editor/extensions/temp-file.tsx
@@ -24,27 +24,6 @@ export const TempFileNode = Node.create({
       id: {
         default: null,
       },
-      name: {
-        default: null,
-      },
-      path: {
-        default: null,
-      },
-      size: {
-        default: null,
-      },
-      type: {
-        default: null,
-      },
-      mimeType: {
-        default: null,
-      },
-      extension: {
-        default: null,
-      },
-      url: {
-        default: null,
-      },
     };
   },
   renderHTML({ HTMLAttributes }) {
@@ -66,7 +45,9 @@ export const TempFileNode = Node.create({
             .focus()
             .insertContentAt(pos, {
               type: 'tempFile',
-              attrs: file,
+              attrs: {
+                id: file.id,
+              },
             })
             .run();