diff --git a/action.yml b/action.yml
index 68750d9..ae28288 100644
--- a/action.yml
+++ b/action.yml
@@ -39,7 +39,7 @@ outputs:
 runs:
   using: 'composite'
   steps:
-    - uses: actions/attest-build-provenance/generate-build-provenance-predicate@input-refactor
+    - uses: actions/attest-build-provenance/generate-build-provenance-predicate@main
       id: generate-build-provenance-predicate
     - uses: actions/attest@main
       id: attest