diff --git a/src/toolbox_scs/routines/boz.py b/src/toolbox_scs/routines/boz.py
index ca0dd878debe73b01311244ef1c3a30c65fe1c92..e5ac8d409cdedb4626860badab4f8ca0f3db0bad 100644
--- a/src/toolbox_scs/routines/boz.py
+++ b/src/toolbox_scs/routines/boz.py
@@ -284,21 +284,28 @@ def _get_pixel_hexagons(module):
     
     Parameters:
     -----------
-    module: int, module number
+    module: either int, for the module number or a 2-d array of corners to
+        get hexagons from
 
     Returns:
-    a list of hexagons
+    --------
+    a 1-d list of hexagons where corners position are in mm
     """
     hexes = []
     
-    corners = _get_pixel_corners(module)
-    for y in range(128):
-        for x in range(512):
+    if type(module) is int:
+        corners = _get_pixel_corners(module)
+    else:
+        corners = module
+
+    for y in range(corners.shape[0]):
+        for x in range(corners.shape[1]):
             c = 1e3*corners[y, x, :, :] # convert to mm
             hexes.append(Polygon(c))
 
     return hexes
 
+
 def _add_colorbar(im, ax, loc='right', size='5%', pad=0.05):
     """Add a colobar on a new axes so it match the plot size.