diff --git a/src/frontend/src/contexts/tabsContext.tsx b/src/frontend/src/contexts/tabsContext.tsx
index a373ad9a7..8be0a7c69 100644
--- a/src/frontend/src/contexts/tabsContext.tsx
+++ b/src/frontend/src/contexts/tabsContext.tsx
@@ -85,6 +85,7 @@ export function TabsProvider({ children }: { children: ReactNode }) {
     if (cookie && Object.keys(templates).length > 0) {
       let cookieObject: LangFlowState = JSON.parse(cookie);
       cookieObject.flows.forEach((flow) => {
+        flow.data.edges.forEach((edge) => {edge.className = "";})
         flow.data.nodes.forEach((node) => {
           if (Object.keys(templates[node.data.type]["template"]).length > 0) {
             node.data.node.base_classes =
@@ -285,8 +286,8 @@ export function TabsProvider({ children }: { children: ReactNode }) {
     // Get data from the flow or set it to null if there's no flow provided.
     const data = flow?.data ? flow.data : null;
     const description = flow?.description ? flow.description : "";
-
     if (data) {
+      data.edges.forEach((edge) => {edge.className = "";})
       data.nodes.forEach((node) => {
         if (Object.keys(templates[node.data.type]["template"]).length > 0) {
           node.data.node.base_classes =