fix

@@ -1396,7 +1396,8 @@ export default {
           control,
           control,
         }),
         }),
       );
       );
-      edges = edges.map(({ source, sourceAnchor, target, targetAnchor }) => ({
+      edges = edges.map(({ shape, source, sourceAnchor, target, targetAnchor }) => ({
+        shape,
         source,
         source,
         sourceAnchor,
         sourceAnchor,
         target,
         target,