     return linkGenerator({
            source: [sourceNode.x + 50, sourceNode.y],
            target: [targetNode.x - 50, targetNode.y],
          });
        });

      // Draw nodes
      const nodeGroups = svg
        .selectAll(".node-group")
        .data(nodes)
        .enter()
        .append("g")
        .attr("transform", (d) => `translate(${d.x},${d.y})`);

      nodeGroups
        .append("rect")
        .attr("class", "node")
        .attr("x", -50)
        .attr("y", -30)
        .attr("width"