diff --git a/src/components/generic/Flow/Edges/BiDirectionalEdge.tsx b/src/components/generic/Flow/Edges/BiDirectionalEdge.tsx
index 7ba014f..3328124 100644
--- a/src/components/generic/Flow/Edges/BiDirectionalEdge.tsx
+++ b/src/components/generic/Flow/Edges/BiDirectionalEdge.tsx
@@ -1,5 +1,5 @@
 import { memo } from 'react';
-import { getBezierPath, BaseEdge, useStore, EdgeProps, ReactFlowState, EdgeLabelRenderer, MarkerType } from 'reactflow';
+import { getBezierPath, BaseEdge, useStore, EdgeProps, ReactFlowState, EdgeLabelRenderer } from 'reactflow';
 
 export type GetSpecialPathParams = {
   sourceX: number;
@@ -30,6 +30,7 @@ export default memo(({
   targetPosition,
   selected,
   data,
+  markerEnd,
 }: EdgeProps) => {
   const isBiDirectionEdge = useStore((s: ReactFlowState) => {
     const edgeExists = s.edges.some(
@@ -61,7 +62,7 @@ export default memo(({
 
   return (
     <>
-      <BaseEdge id={id} path={path} markerEnd={MarkerType.Arrow} style={{
+      <BaseEdge id={id} path={path} markerEnd={markerEnd} style={{
         strokeWidth: 2,
         zIndex: 1,
         stroke: selected ? '#FF0072' : '#000',
diff --git a/src/components/generic/Flow/index.tsx b/src/components/generic/Flow/index.tsx
index 9d56ee0..23bd33d 100644
--- a/src/components/generic/Flow/index.tsx
+++ b/src/components/generic/Flow/index.tsx
@@ -13,6 +13,7 @@ import ReactFlow, {
   useReactFlow,
   applyEdgeChanges,
   applyNodeChanges,
+  MarkerType,
 } from 'reactflow';
 import styled from '@emotion/styled';
 import Container from '@cloudscape-design/components/container';
@@ -177,7 +178,7 @@ function Flow() {
           selectedThreats: [],
         },
       };
-      setEdges((eds) => addEdge(newEdge, eds));
+      setEdges((eds) => addEdge(addEndMarker(newEdge), eds));
       setNodeDataValue({});
     }, [setEdges],
   );
@@ -217,6 +218,15 @@ function Flow() {
     setThreatList(statementList);
   }, [setThreatList, statementList]);
 
+  const addEndMarker = (edge) => ({
+    ...edge,
+    markerEnd: {
+      type: MarkerType.ArrowClosed,
+      width: 20,
+      height: 20,
+      color: '#b1b1b7',
+    },
+  });
 
   return (
     <SpaceBetween direction="vertical" size="s">