diff --git a/deploy/python/picodet_postprocess.py b/deploy/python/picodet_postprocess.py
index 1fed8c6411c02b502399963811de516538d802d8..f5911a697b5f7a25db61a1b42351258341fe548b 100644
--- a/deploy/python/picodet_postprocess.py
+++ b/deploy/python/picodet_postprocess.py
@@ -194,7 +194,7 @@ class PicoDetPostProcess(object):
                 picked_box_probs.append(box_probs)
                 picked_labels.extend([class_index] * box_probs.shape[0])
             if not picked_box_probs:
-                return np.array([]), np.array([]), np.array([])
+                return np.array([]), np.array([])
             picked_box_probs = np.concatenate(picked_box_probs)
 
             # resize output boxes