diff --git a/tests/test_tbx11k.py b/tests/test_tbx11k.py
index a4c31edb5cb0c1436ea7c4a71f9602ce9fa7b5d1..e7dae5678b1cc3ff18cc728a485f33d730de24f6 100644
--- a/tests/test_tbx11k.py
+++ b/tests/test_tbx11k.py
@@ -193,19 +193,19 @@ def check_loaded_batch(
     ):
         # there must be a sign indicated on the image, if active TB is detected
         if label == 1:
-            assert len(bboxes[0]) != 0
+            assert len(bboxes) != 0
 
         # eif label == 0:  # not true, may have TBI!
         #    assert len(bboxes) == 0
 
         # asserts all bounding boxes are within the raw image width and height
-        for bbox_label, xmin, ymin, width, height in zip(*bboxes):
+        for bbox in bboxes:
             if label == 1:
-                assert bbox_label == 1
+                assert bbox.label == 1
             else:
-                assert bbox_label == 0
-            assert (xmin + width) < sample.shape[2]
-            assert (ymin + height) < sample.shape[1]
+                assert bbox.label == 0
+            assert bbox.xmax < sample.shape[2]
+            assert bbox.ymax < sample.shape[1]
 
     # use the code below to view generated images
     # from torchvision.transforms.functional import to_pil_image