diff --git a/docker/docker-compose.al2.54.yaml b/docker/docker-compose.al2.54.yaml
index c6b2a7ac..cb300eb7 100644
--- a/docker/docker-compose.al2.54.yaml
+++ b/docker/docker-compose.al2.54.yaml
@@ -6,7 +6,7 @@ services:
     image: swift-aws-lambda:al2-5.4
     build:
       args:
-        base_image: "swiftlang/swift:nightly-5.4-amazonlinux2"
+        swift_version: "5.4"
 
   test:
     image: swift-aws-lambda:al2-5.4