diff --git a/python/caffe/layers/detectnet/clustering.py b/python/caffe/layers/detectnet/clustering.py
index 9547db7d880..77bec07387b 100644
--- a/python/caffe/layers/detectnet/clustering.py
+++ b/python/caffe/layers/detectnet/clustering.py
@@ -150,7 +150,8 @@ def gridbox_to_boxes(net_cvg, net_boxes, self):
     cell_width = im_sz_x / grid_sz_x
     cell_height = im_sz_y / grid_sz_y
 
-    cvg_val = net_cvg[0, 0:grid_sz_y, 0:grid_sz_x]
+    #cvg_val = net_cvg[0, 0:grid_sz_y, 0:grid_sz_x]
+    cvg_val = net_cvg[0][0:net_cvg.shape[1],0:net_cvg.shape[2]]
 
     if (self.is_groundtruth):
         mask = (cvg_val > 0)