diff --git a/yolov5.py b/yolov5.py
index 9e927f9..477d9d6 100644
--- a/yolov5.py
+++ b/yolov5.py
@@ -14,7 +14,7 @@ class Yolov5(NNOperator):
         # Get object detection results with YOLOv5 model
         results = self._model(img)
 
-        boxes = [re[0:3] for re in results.xyxy[0]]
+        boxes = [re[0:4] for re in results.xyxy[0]]
         boxes = [list(map(int, box)) for box in boxes]
         classes = list(results.pandas().xyxy[0].name)
         scores = list(results.pandas().xyxy[0].confidence)