diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py
index 4efb80b396da757b2f9e48f5b3061a920f290f2e..c7813d52dd90061c38f5dfacdef446c78df7de01 100644
--- a/model_server/base/roiset.py
+++ b/model_server/base/roiset.py
@@ -229,8 +229,14 @@ class RoiSet(object):
         zmask = np.zeros((*acc_raw.hw, 1, acc_raw.nz), dtype=bool)
 
         def _label_obj(r):
-            sl = np.s_[r.ebb_y0:r.ebb_y1, r.ebb_x0:r.ebb_x1, :, r.zi:r.zi + 1]
-            zmask[sl] = patch_masks[r.label]
+            # sl = np.s_[r.ebb_y0:r.ebb_y1, r.ebb_x0:r.ebb_x1, :, r.zi:r.zi + 1]
+            sl = np.s_[r.y0:r.y1, r.x0:r.x1, :, r.zi:r.zi + 1]
+            rsl = np.s_[r.rel_y0:r.rel_y1, r.rel_x0:r.rel_x1, :, :]
+            try:
+                zmask[sl] = patch_masks[r.label][rsl]
+                print(f'ebb (h, w) = {r.ebb_h, r.ebb_w}')
+            except ValueError as ve:
+                print(ve)
 
         df.apply(lambda x: _label_obj(x), axis=1)
         obj_ids = _get_label_ids(zmask, allow_3d=True, connect_3d=False)