diff --git a/tests/integ/test_models_from_hub.py b/tests/integ/test_models_from_hub.py
index 3125c1c..8b09ce5 100644
--- a/tests/integ/test_models_from_hub.py
+++ b/tests/integ/test_models_from_hub.py
@@ -38,8 +38,8 @@ def get_all_ecr_images(registry_id, repository_name, result_key):
     image_tags = [image["imageTag"] for image in images]
     print(image_tags)
     image_regex = re.compile("\d\.\d\.\d-" + device + "-.{4}$")
-    image = sorted(list(filter(image_regex.match, image_tags)), reverse=True)[0]
-    return image
+    tag = sorted(list(filter(image_regex.match, image_tags)), reverse=True)[0]
+    return f"{registry_id}.dkr.ecr.{os.environ.get('AWS_DEFAULT_REGION','us-east-1')}.amazonaws.com/{repository_name}:{tag}"
 
 
 @pytest.mark.parametrize(