diff --git a/templates/docker-compose/template.yml b/templates/docker-compose/template.yml
index f9a0f2f1e66ea45a77e2641b3e1d65b73c42a53a..1303b0c779618b4b69e7e8f2008ece22c6ef9ae6 100644
--- a/templates/docker-compose/template.yml
+++ b/templates/docker-compose/template.yml
@@ -6,7 +6,7 @@ spec:
 docker-compose-build:
     # Use the official docker image.
     image: docker:27
-    stage: build
+    stage: $[[ inputs.stage ]]
     services:
       - name: docker:27-dind
         alias: docker