diff --git a/app/setup.py b/app/setup.py
index 791933c..8212b39 100644
--- a/app/setup.py
+++ b/app/setup.py
@@ -136,11 +136,9 @@ base_stub = Image.from_dockerfile(
     path="Dockerfile",
     context_mount=Mount.from_local_file("requirements.txt"),
 )
-stub.image = base_stub.extend(
-    dockerfile_commands=[
-        "FROM base",
-        "COPY config.yml /",
-    ],
+stub.image = base_stub.dockerfile_commands(
+    "FROM base",
+    "COPY config.yml /",
     context_mount=Mount.from_local_file("config.yml"),
 ).run_function(
     build_image,