diff --git a/models/common.py b/models/common.py
index 83cc8b5ce27b..721171393e04 100644
--- a/models/common.py
+++ b/models/common.py
@@ -235,7 +235,7 @@ def forward(self, imgs, size=640, augment=False, profile=False):
 
 class Detections:
     # detections class for YOLOv5 inference results
-    def __init__(self, imgs, pred, files, times, names=None, shape=None):
+    def __init__(self, imgs, pred, files, times=None, names=None, shape=None):
         super(Detections, self).__init__()
         d = pred[0].device  # device
         gn = [torch.tensor([*[im.shape[i] for i in [1, 0, 1, 0]], 1., 1.], device=d) for im in imgs]  # normalizations