diff --git a/tests/e2e/cli/detection/test_api_xai_sanity_detection.py b/tests/e2e/cli/detection/test_api_xai_sanity_detection.py
index 02024d54bfd..4cd11fa937c 100644
--- a/tests/e2e/cli/detection/test_api_xai_sanity_detection.py
+++ b/tests/e2e/cli/detection/test_api_xai_sanity_detection.py
@@ -32,7 +32,7 @@
 
 class TestOVDetXAIAPI(DetectionTaskAPIBase):
     ref_raw_saliency_shapes = {
-        "MobileNetV2-ATSS": (4, 4),  # Need to be adapted to configurable or adaptive input size
+        "MobileNetV2-ATSS": (16, 16),  # Need to be adapted to configurable or adaptive input size
     }
 
     @e2e_pytest_api