diff --git a/workflow_indoor.py b/workflow_indoor.py
index 71ef432..129681d 100644
--- a/workflow_indoor.py
+++ b/workflow_indoor.py
@@ -271,6 +271,7 @@ class IndoorNavigator:
                 masks = r0.masks.data.cpu().numpy()
                 boxes = r0.boxes
                 
+                for i, (mask, cls_id, conf) in enumerate(zip(masks, boxes.cls, boxes.conf)):
                     cls_id = int(cls_id.item())
                     conf_val = float(conf.item())