diff --git a/src/preprocessor/soundpreprocessor.py b/src/preprocessor/soundpreprocessor.py
index 28744ad..b43b3fa 100644
--- a/src/preprocessor/soundpreprocessor.py
+++ b/src/preprocessor/soundpreprocessor.py
@@ -130,7 +130,7 @@ class SoundPreProcessor(AbcPreProcessor):
             f"Sample {file_path} identified as {self._classes[class_id]} with the probablility of {probability[class_id]}"
         )
 
-        return bool((class_id == self._target_id) and (probability > 0.5))
+        return bool((class_id == self._target_id) and (probability[class_id] > 0.5))
 
     def __del__(self):
         try: