diff --git a/templates/nix-gitlab-ci.yml b/templates/nix-gitlab-ci.yml
index 2dcead3..cc87788 100644
--- a/templates/nix-gitlab-ci.yml
+++ b/templates/nix-gitlab-ci.yml
@@ -23,18 +23,10 @@ spec:
       type: string
       description: |
         Which version of the Nix CI image to use. Using a tag/version is recommended.
-    stage_build:
-      type: string
-      description: The CI stage for building the dynamic pipeline.
-      default: build
-    stage_trigger:
-      type: string
-      description: The CI stage for triggering the dynamic pipeline.
-      default: trigger
 ---
 stages:
-  - $[[ inputs.stage_build ]]
-  - $[[ inputs.stage_trigger ]]
+  - build
+  - trigger
 variables:
   # These can be overriden, see https://docs.gitlab.com/ci/variables/#cicd-variable-precedence
   # which image should be used by default.
@@ -42,7 +34,7 @@ variables:
   # default cache stategy
   NIX_CI_CACHE_STRATEGY: $[[ inputs.cache_strategy ]]
 nix-ci:build:
-  stage: $[[ inputs.stage_build ]]
+  stage: build
   image: $NIX_CI_IMAGE
   cache:
     - key:
@@ -100,7 +92,7 @@ nix-ci:build:
       dotenv: trigger.env
 
 nix-ci:trigger:
-  stage: $[[ inputs.stage_trigger ]]
+  stage: trigger
   needs:
     - nix-ci:build
   trigger: