Merge pull request #495 from wbkd/develop

Develop

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}

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;