diff --git a/motrackers/detectors/yolo.py b/motrackers/detectors/yolo.py
index d6fd24d..894e911 100644
--- a/motrackers/detectors/yolo.py
+++ b/motrackers/detectors/yolo.py
@@ -23,10 +23,10 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh
         object_names = load_labelsjson(labels_path)
 
         layer_names = self.net.getLayerNames()
-        if cv2.__version__ == '4.6.0':
+        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)