diff --git a/wait-for-ecr-scan-and-get-sarif/main.py b/wait-for-ecr-scan-and-get-sarif/main.py
index e4fea9c..8f1594a 100644
--- a/wait-for-ecr-scan-and-get-sarif/main.py
+++ b/wait-for-ecr-scan-and-get-sarif/main.py
@@ -62,7 +62,7 @@ def convert_to_sarif(ecr_response):
                 {
                     "physicalLocation": {
                         "artifactLocation": {
-                            "uri": "library/ubuntu",
+                            "uri": ecr_response["repositoryName"],
                             "uriBaseId": "ROOTPATH"
                         },
                         "region": {
@@ -73,7 +73,7 @@ def convert_to_sarif(ecr_response):
                         }
                     },
                     "message": {
-                        "text": f"library/ubuntu: {finding['attributes'][1]['value']}@{finding['attributes'][0]['value']}"
+                        "text": f"{ecr_response['repositoryName']}: {finding['attributes'][1]['value']}@{finding['attributes'][0]['value']}"
                     }
                 }
             ]