diff --git a/rknn.cpp b/rknn.cpp
index 31611ac..244b52d 100644
--- a/rknn.cpp
+++ b/rknn.cpp
@@ -119,7 +119,7 @@ int print_object_detect_result(rknn_app_context_t *rknn, object_detect_result_li
             continue;
         }
         char *label = labels[group->results[i].cls_id];
-        if (group->results[i].cls_id != 5 && group->results[i].prop < conf_threshold) {
+        if (group->results[i].cls_id != 4 && group->results[i].prop < conf_threshold) {
             printf("skip prop = %f for class %s\n", group->results[i].prop, label);
             continue;
         }