diff --git a/myeasymocap/backbone/yolo/yolo.py b/myeasymocap/backbone/yolo/yolo.py
index 3567e44..74ff54c 100644
--- a/myeasymocap/backbone/yolo/yolo.py
+++ b/myeasymocap/backbone/yolo/yolo.py
@@ -90,7 +90,7 @@ class BaseYOLOv5:
         select = np.stack(select)
         return select, results
 
-    def select_bbox(self, select, imgname):
+    def select_bbox(self, select, results, imgname):
         if select.shape[0] == 0:
             return select
         # Naive: select the best