From fa193bddfc921a00df5d476e5ee47708664d46fc Mon Sep 17 00:00:00 2001
From: Stojan Stamkov <stamkovs@yahoo.com>
Date: Fri, 4 Sep 2020 21:04:53 +0200
Subject: [PATCH 1/2] Added support for Node and Edge type updates

---
 src/hooks/useElementUpdater.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/hooks/useElementUpdater.ts b/src/hooks/useElementUpdater.ts
index 22a3c156..f0109bf5 100644
--- a/src/hooks/useElementUpdater.ts
+++ b/src/hooks/useElementUpdater.ts
@@ -41,6 +41,10 @@ const useElementUpdater = (propElements: Elements): void => {
         if (typeof propElement.isHidden !== 'undefined') {
           elementProps.isHidden = propElement.isHidden;
         }
+        
+        if (typeof propElement.type !== 'undefined') {
+          elementProps.type = propElement.type;
+        }
 
         if (isNode(existingElement)) {
           const propNode = propElement as Node;

From e06ffe8b9a4aed4c91db308bfeddecba7890a06e Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Sat, 5 Sep 2020 12:00:09 +0200
Subject: [PATCH 2/2] refactor(draggable): pass nodeRef to draggable comp
 closes #486

---
 src/components/Nodes/wrapNode.tsx | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/components/Nodes/wrapNode.tsx b/src/components/Nodes/wrapNode.tsx
index c4f6ea86..c43a3c6a 100644
--- a/src/components/Nodes/wrapNode.tsx
+++ b/src/components/Nodes/wrapNode.tsx
@@ -324,6 +324,7 @@ export default (NodeComponent: ComponentType<NodeComponentProps>) => {
         scale={transform[2]}
         disabled={!isDraggable}
         cancel=".nodrag"
+        nodeRef={nodeElement}
       >
         <div
           className={nodeClasses}