diff --git a/action.yml b/action.yml
index c16174a..5abd2b4 100644
--- a/action.yml
+++ b/action.yml
@@ -19,3 +19,5 @@ runs:
       node-version: 16
   - run: node dist/index.js
     shell: bash
+    env:
+      INPUT_DAFNY_VERSION: ${{ inputs.dafny-version }}