diff --git a/docker/docker-compose.al2.54.yaml b/docker/docker-compose.al2.54.yaml
new file mode 100644
index 00000000..c6b2a7ac
--- /dev/null
+++ b/docker/docker-compose.al2.54.yaml
@@ -0,0 +1,18 @@
+version: "3"
+
+services:
+
+  runtime-setup:
+    image: swift-aws-lambda:al2-5.4
+    build:
+      args:
+        base_image: "swiftlang/swift:nightly-5.4-amazonlinux2"
+
+  test:
+    image: swift-aws-lambda:al2-5.4
+
+  test-samples:
+    image: swift-aws-lambda:al2-5.4
+
+  shell:
+    image: swift-aws-lambda:al2-5.4
diff --git a/docker/docker-compose.al2.main.yaml b/docker/docker-compose.al2.main.yaml
new file mode 100644
index 00000000..c63a13e2
--- /dev/null
+++ b/docker/docker-compose.al2.main.yaml
@@ -0,0 +1,18 @@
+version: "3"
+
+services:
+
+  runtime-setup:
+    image: swift-aws-lambda:al2-main
+    build:
+      args:
+        base_image: "swiftlang/swift:nightly-main-amazonlinux2"
+
+  test:
+    image: swift-aws-lambda:al2-main
+
+  test-samples:
+    image: swift-aws-lambda:al2-main
+
+  shell:
+    image: swift-aws-lambda:al2-main
diff --git a/docker/docker-compose.yaml b/docker/docker-compose.yaml
index 7feae683..122ccd9d 100644
--- a/docker/docker-compose.yaml
+++ b/docker/docker-compose.yaml
@@ -1,6 +1,6 @@
 # this file is not designed to be run directly
 # instead, use the docker-compose.<os>.<swift> files
-# eg docker-compose -f docker/docker-compose.yaml -f docker/docker-compose.1804.52.yaml run test
+# eg docker-compose -f docker/docker-compose.yaml -f docker/docker-compose.al2.52.yaml run test
 version: "3"
 
 services: