diff --git a/.gitattributes b/.gitattributes
index 379b21be8..c79a33b70 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -32,4 +32,4 @@ Dockerfile text
 *.mp4 binary
 *.svg binary
 *.csv binary
-
+*.wav binary
diff --git a/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx b/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx
index 4e60aeeb7..2055cfa05 100644
--- a/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx
@@ -51,24 +51,19 @@ export default function UpdateAllComponents({}: {}) {
         title: ERROR_MESSAGE_EDGES_LOST,
       });
 
-      edgesUpdateRef.current = {
-        numberOfEdgesBeforeUpdate: 0,
-        updateComponent: false,
-      };
+      resetEdgesUpdateRef();
     }
   }, [edges]);
 
   const getSuccessTitle = (updatedCount: number) => {
+    resetEdgesUpdateRef();
     return `Successfully updated ${updatedCount} component${
       updatedCount > 1 ? "s" : ""
     }`;
   };
 
   const handleUpdateAllComponents = () => {
-    edgesUpdateRef.current = {
-      numberOfEdgesBeforeUpdate: edges.length,
-      updateComponent: true,
-    };
+    startEdgesUpdateRef();
 
     setLoadingUpdate(true);
     takeSnapshot();
@@ -135,6 +130,20 @@ export default function UpdateAllComponents({}: {}) {
       });
   };
 
+  const resetEdgesUpdateRef = () => {
+    edgesUpdateRef.current = {
+      numberOfEdgesBeforeUpdate: 0,
+      updateComponent: false,
+    };
+  };
+
+  const startEdgesUpdateRef = () => {
+    edgesUpdateRef.current = {
+      numberOfEdgesBeforeUpdate: edges.length,
+      updateComponent: true,
+    };
+  };
+
   if (componentsToUpdate.length === 0) return null;
 
   return (