diff --git a/action.yml b/action.yml
index 85da2f9..2bcf1ff 100644
--- a/action.yml
+++ b/action.yml
@@ -18,4 +18,4 @@ inputs:
     required: false
 runs:
   using: 'node12'
-  main: 'lib/setup-java.js'
+  main: 'dist/index.js'