diff --git a/ts/torch_handler/unit_tests/test_object_detector.py b/ts/torch_handler/unit_tests/test_object_detector.py
index 2693e66673..233dc03bd0 100644
--- a/ts/torch_handler/unit_tests/test_object_detector.py
+++ b/ts/torch_handler/unit_tests/test_object_detector.py
@@ -33,4 +33,4 @@ def test_handle(model_setup):
     test_data = [{'data': image_bytes}] * 2
     results = handler.handle(test_data, context)
     assert(len(results) == 2)
-    assert('bench' in results[0])
+    assert(any('bench' in d for d in results[0]))