diff --git a/lib/osml/osml_container.ts b/lib/osml/osml_container.ts
index 77008d1..9fbe859 100644
--- a/lib/osml/osml_container.ts
+++ b/lib/osml/osml_container.ts
@@ -268,7 +268,9 @@ export class OSMLContainer extends Construct {
 
       if (this.buildDockerImageCode) {
         // Create a DockerImageCode object for Lambda using the imported ECR repository.
-        this.dockerImageCode = DockerImageCode.fromEcr(this.repository);
+        this.dockerImageCode = DockerImageCode.fromEcr(this.repository, {
+          tagOrDigest: extractedTag
+        });
       }
     } else {
       // If the CONTAINER_URI does not indicate an ECR repository, assume it is a public or private Docker registry.