diff --git a/utils/utils.py b/utils/utils.py
index 10502174..a9f77a1c 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -688,7 +688,7 @@ def kmeans_targets(path='../coco/trainvalno5k.txt', n=9, img_size=416):  # from
     wh = np.concatenate(dataset.labels, 0)[:, 3:5]  # wh from cxywh
 
     # Kmeans calculation
-    k = cluster.vq.kmeans(wh, n)[0]
+    k, dist = cluster.vq.kmeans(wh, n)  # points, mean distance
     k = k[np.argsort(k.prod(1))]  # sort small to large
 
     # Measure IoUs