diff --git a/motrackers/detectors/yolo.py b/motrackers/detectors/yolo.py
index ee169ec..05d3d7d 100644
--- a/motrackers/detectors/yolo.py
+++ b/motrackers/detectors/yolo.py
@@ -26,7 +26,7 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh
         if cv.__version__ >= '4.6.0':
             self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
         else:
-            self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()]
+            self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
 
         self.scale_factor = 1/255.0
         self.image_size = (416, 416)