diff --git a/docker/docker-compose.2004.54.yaml b/docker/docker-compose.2004.54.yaml
index dcea5ae..d9c441b 100644
--- a/docker/docker-compose.2004.54.yaml
+++ b/docker/docker-compose.2004.54.yaml
@@ -6,7 +6,8 @@ services:
     image: swift-metrics:20.04-5.4
     build:
       args:
-        base_image: "swiftlang/swift:nightly-5.4-focal"
+        ubuntu_version: "focal"
+        swift_version: "5.4"
 
   test:
     image: swift-metrics:20.04-5.4