diff --git a/src/components/AnnotationMain.js b/src/components/AnnotationMain.js
index 002d30e..6bc4ff7 100644
--- a/src/components/AnnotationMain.js
+++ b/src/components/AnnotationMain.js
@@ -42,7 +42,6 @@ function AnnotationMain() {
 
     const handleAnnotationSave = () => {
         const containerRef = { current: { offsetWidth: videoRef.current.videoWidth, offsetHeight: videoRef.current.videoHeight } };
-        const annotationSelectedClass = selectedClass;
         const imageData = AnnotationService.createAnnotationImage(videoRef, detections, null, annotationSelectedClass, containerRef);
         if (imageData) {
             setAnnotations(prevAnnotations => ({
diff --git a/src/components/DetectionContainer.js b/src/components/DetectionContainer.js
index a449f84..8d20b82 100644
--- a/src/components/DetectionContainer.js
+++ b/src/components/DetectionContainer.js
@@ -2,7 +2,7 @@
 import React from 'react';
 import Detection from './Detection';
 
-function DetectionContainer({ detections, selectedDetectionIndices, calculateColor, onDetectionMouseDown, currentDetection, onResize }) {
+function DetectionContainer({ detections, selectedDetectionIndices, onDetectionMouseDown, currentDetection, onResize }) {
 
     return (
         <>