diff --git a/utils/utils.py b/utils/utils.py
index 121f352e..bccb4278 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -99,7 +99,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
     tp, conf, pred_cls = tp[i], conf[i], pred_cls[i]
 
     # Find unique classes
-    unique_classes = np.unique(np.concatenate((pred_cls, target_cls), 0))
+    unique_classes = target_cls #np.unique(np.concatenate((pred_cls, target_cls), 0))
 
     # Create Precision-Recall curve and compute AP for each class
     ap = []