diff --git a/resources/js/wysiwyg/lexical/rich-text/LexicalDiagramNode.ts b/resources/js/wysiwyg/lexical/rich-text/LexicalDiagramNode.ts
index e69f97848..6be19c293 100644
--- a/resources/js/wysiwyg/lexical/rich-text/LexicalDiagramNode.ts
+++ b/resources/js/wysiwyg/lexical/rich-text/LexicalDiagramNode.ts
@@ -27,7 +27,7 @@ export class DiagramNode extends DecoratorNode<EditorDecoratorAdapter> {
     }
 
     static clone(node: DiagramNode): DiagramNode {
-        const newNode = new DiagramNode(node.__drawingId, node.__drawingUrl);
+        const newNode = new DiagramNode(node.__drawingId, node.__drawingUrl, node.__key);
         newNode.__id = node.__id;
         return newNode;
     }
diff --git a/resources/js/wysiwyg/ui/decorators/DiagramDecorator.ts b/resources/js/wysiwyg/ui/decorators/DiagramDecorator.ts
index e46dcc312..58a4de1dc 100644
--- a/resources/js/wysiwyg/ui/decorators/DiagramDecorator.ts
+++ b/resources/js/wysiwyg/ui/decorators/DiagramDecorator.ts
@@ -1,5 +1,4 @@
 import {EditorDecorator} from "../framework/decorator";
-import {EditorUiContext} from "../framework/core";
 import {BaseSelection, CLICK_COMMAND, COMMAND_PRIORITY_NORMAL} from "lexical";
 import {DiagramNode} from "@lexical/rich-text/LexicalDiagramNode";
 import {$selectionContainsNode, $selectSingleNode} from "../../utils/selection";