diff --git a/src/components/AnnotationMain/AnnotationMain.js b/src/components/AnnotationMain/AnnotationMain.js
index dafade7..79e215b 100644
--- a/src/components/AnnotationMain/AnnotationMain.js
+++ b/src/components/AnnotationMain/AnnotationMain.js
@@ -75,17 +75,15 @@ function AnnotationMain() {
         );
 
         if (imageData) {
-            setAnnotations(prevAnnotations => {
-                const newAnnotations = {
-                    ...prevAnnotations,
-                    [currentTime]: { time: currentTime, annotations: detections, imageData }
-                };
-
-                saveAnnotation(currentTime, detections, imageData);
-                setErrorMessage("");
-
-                return newAnnotations;
-            });
+            const newAnnotations = {
+                ...annotations,
+                [currentTime]: { time: currentTime, annotations: detections, imageData }
+            };
+        
+            setAnnotations(newAnnotations);
+        
+            saveAnnotation(currentTime, detections, imageData);
+            setErrorMessage("");
         }
     };
 
@@ -132,7 +130,7 @@ function AnnotationMain() {
     };
 
     const handleStop = () => {
-        setIsPlaying(prev => !prev);
+        setIsPlaying(false);
         setCurrentTime(0);
     };