diff --git a/src/frontend/src/App.css b/src/frontend/src/App.css
index 809959757..5a97d371e 100644
--- a/src/frontend/src/App.css
+++ b/src/frontend/src/App.css
@@ -174,3 +174,12 @@ body {
   border: none !important;
   outline: none;
 }
+
+/* selected */
+.react-flow__edge.selected .react-flow__edge-path {
+  stroke: var(--selected) !important;
+}
+
+.react-flow__edge .react-flow__edge-path {
+  stroke: var(--connection) !important;
+}
diff --git a/src/frontend/src/stores/flowStore.ts b/src/frontend/src/stores/flowStore.ts
index d966868a2..99aea0515 100644
--- a/src/frontend/src/stores/flowStore.ts
+++ b/src/frontend/src/stores/flowStore.ts
@@ -315,8 +315,6 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
           targetHandle,
           id,
           data: cloneDeep(edge.data),
-          style: { stroke: "#555" },
-          className: "stroke-gray-900 ",
           selected: false,
         },
         newEdges.map((edge) => ({ ...edge, selected: false })),
@@ -396,8 +394,8 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
             targetHandle: scapeJSONParse(connection.targetHandle!),
             sourceHandle: scapeJSONParse(connection.sourceHandle!),
           },
-          style: { stroke: "#555" },
-          className: "stroke-foreground stroke-connection",
+          // style: { stroke: "#555" },
+          // className: "stroke-foreground stroke-connection",
         },
         oldEdges,
       );
diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index a1e592458..161f7c70b 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -179,7 +179,7 @@ export const processFlows = (DbData: FlowType[], skipUpdate = true) => {
         ] = cloneDeep((flow.data.nodes[0].data as NodeDataType).node!);
         return;
       }
-      if (!skipUpdate) processDataFromFlow(flow, false);
+      processDataFromFlow(flow, !skipUpdate);
     } catch (e) {
       console.log(e);
     }
@@ -345,7 +345,7 @@ export function updateEdges(edges: Edge[]) {
       const targetHandleObject: targetHandleType = scapeJSONParse(
         edge.targetHandle!,
       );
-      edge.className = "stroke-gray-900 stroke-connection";
+      edge.className = "";
     });
 }
 
@@ -1005,11 +1005,6 @@ export function processFlowEdges(flow: FlowType) {
     const newEdges = updateEdgesHandleIds(flow.data);
     flow.data.edges = newEdges;
   }
-  //update edges colors
-  flow.data.edges.forEach((edge) => {
-    edge.className = "";
-    edge.style = { stroke: "#555" };
-  });
 }
 
 export function expandGroupNode(