diff --git a/model_server/extensions/chaeo/zmask.py b/model_server/extensions/chaeo/zmask.py
index f77ac431ab9cd5fc67f6ef5a83809e5153df75da..587c801b256c15c1975cb301e85596e2e6ce3062 100644
--- a/model_server/extensions/chaeo/zmask.py
+++ b/model_server/extensions/chaeo/zmask.py
@@ -292,9 +292,9 @@ def build_zmask_from_object_mask(
             'df_index': ob.Index,
             'info': ob,
             'slice': sl,
-            'relative_bounding_box': rbb,
-            'contour': contour,
-            'mask': mask
+            'relative_bounding_box': rbb, # TODO: put in DF
+            'contour': contour, # TODO: delegate to getter
+            'mask': mask # TODO: delegate to getter
         })
 
     # return intermediate image arrays