diff --git a/action.yml b/action.yml
index e6bf35f..65c4eee 100644
--- a/action.yml
+++ b/action.yml
@@ -6,9 +6,6 @@ inputs:
   hax_repository:
     required: true
     default: "github:hacspec/hax"
-  hax_flake_uri:
-    required: true
-    default: "${{ inputs.hax_repository }}/${{ inputs.hax_reference }}"  
 runs:
   using: "composite"
   steps:
@@ -22,7 +19,7 @@ runs:
     - name: Install binaries
       run: nix profile install "$FLAKE"#hax "$FLAKE"#hax-env "$FLAKE"#fstar
       env:
-        FLAKE: ${{ inputs.hax_flake_uri }}
+        FLAKE: "${{ inputs.hax_repository }}/${{ inputs.hax_reference }}"
 
     - name: Setup environment variables
       run: eval $(hax-env)