diff --git a/src/NetworkVisualizer/Edges.js b/src/NetworkVisualizer/Edges.js
index 8004e26..434255d 100644
--- a/src/NetworkVisualizer/Edges.js
+++ b/src/NetworkVisualizer/Edges.js
@@ -4,15 +4,24 @@ function Edges({ width, height, edges }) {
   return (
     <Stage width={width} height={height}>
       <Layer>
-        {edges.map((edge, i) => (
-          <Arrow
+        {edges.map((edge, i) => {
+          const lastPointIndex = edge.points.length - 1;
+          const arrowStartPoint = edge.points[0];
+          const arrowEndPoint = edge.points[lastPointIndex];
+
+          return <Arrow
             key={i}
-            points={edge.points}
+            points={[
+              arrowStartPoint.x,
+              arrowStartPoint.y,
+              arrowEndPoint.x,
+              arrowEndPoint.y
+            ]}
             strokeWidth={1}
             stroke={"black"}
             fill={"black"}
           />
-        ))}
+        })}
       </Layer>
     </Stage>
   );