diff --git a/kmeans.py b/kmeans.py
index 7cb17e58a8dfd0d1a040bf72b9a3d8736a0ee166..9d823bc69a3dc505e529a66f0178265b1427db68 100644
--- a/kmeans.py
+++ b/kmeans.py
@@ -42,7 +42,7 @@ def translate_boxes(boxes):
     for row in range(new_boxes.shape[0]):
         new_boxes[row][2] = np.abs(new_boxes[row][2] - new_boxes[row][0])
         new_boxes[row][3] = np.abs(new_boxes[row][3] - new_boxes[row][1])
-    return np.delete(boxes, [0, 1], axis=1)
+    return np.delete(new_boxes, [0, 1], axis=1)
 
 
 def kmeans(boxes, k, dist=np.median):