diff --git a/lambda/Dockerfile b/lambda/Dockerfile
index 1e6c581d..ca1adc43 100644
--- a/lambda/Dockerfile
+++ b/lambda/Dockerfile
@@ -1,7 +1,7 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
-ARG buildImage=golang:1
-FROM ${buildImage} as build
+ARG buildImage=public.ecr.aws/docker/library/golang:1
+FROM ${buildImage} AS build
 
 USER root
 
diff --git a/test/lambda/Dockerfile b/test/lambda/Dockerfile
index 51fa4752..84d32e5a 100644
--- a/test/lambda/Dockerfile
+++ b/test/lambda/Dockerfile
@@ -1,7 +1,7 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
-ARG buildImage=golang:1
-FROM ${buildImage} as build
+ARG buildImage=public.ecr.aws/docker/library/golang:1
+FROM ${buildImage} AS build
 
 USER root