diff --git a/utils/utils.py b/utils/utils.py
index b6b15b84..820bd936 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -564,6 +564,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T
             i = iou.max(0)[0] < iou_thres
 
         output[xi] = x[i]
+    return output
 
 
 def get_yolo_layers(model):