diff --git a/infra/template.yml b/infra/template.yml
index 0462fcd..b9bb6d5 100644
--- a/infra/template.yml
+++ b/infra/template.yml
@@ -312,7 +312,7 @@ Resources:
         Type: NO_ARTIFACTS
       Environment:
         # architecture should match the architecture in UserLambdaConfig
-        Type: LINUX_ARM_CONTAINER
+        Type: ARM_CONTAINER
         ComputeType: BUILD_GENERAL1_SMALL
         Image: "aws/codebuild/amazonlinux-aarch64-standard:2.0"
         PrivilegedMode: true # required for Docker builds