diff --git a/packages/core/src/extensions/Blocks/nodes/Block.ts b/packages/core/src/extensions/Blocks/nodes/Block.ts
index bad1b6a833..9cec48364b 100644
--- a/packages/core/src/extensions/Blocks/nodes/Block.ts
+++ b/packages/core/src/extensions/Blocks/nodes/Block.ts
@@ -103,6 +103,7 @@ export const Block = Node.create<IBlock>({
     return [
       {
         tag: "div",
+        context: "blockgroup/",
       },
     ];
   },
diff --git a/packages/core/src/extensions/Blocks/nodes/Content.ts b/packages/core/src/extensions/Blocks/nodes/Content.ts
index 61bff025a9..5c318646a8 100644
--- a/packages/core/src/extensions/Blocks/nodes/Content.ts
+++ b/packages/core/src/extensions/Blocks/nodes/Content.ts
@@ -32,6 +32,7 @@ export const ContentBlock = Node.create<IBlock>({
     return [
       {
         tag: "div",
+        context: "block/",
       },
     ];
   },