diff --git a/unet/segment.py b/unet/segment.py
index 2ca26702520d45e49c8565dcd6677af8bc2f7943..d304893b3451d393ff51b583a551f4deb803085e 100644
--- a/unet/segment.py
+++ b/unet/segment.py
@@ -146,6 +146,5 @@ def get_bounding_box(im):
     """Returns bounding box of object in boolean image"""
     coords = np.where(im)
     
-    min0, min1 = coords.min(axis=0)
-    max0, max1 = coords.max(axis=0)
-    return np.array([min0, max0, min1, max1])
+    return np.array([np.min(coords[0]), np.max(coords[0]), 
+                     np.min(coords[1]), np.max(coords[1])])