diff --git a/action.yml b/action.yml
index e1684fe..eb719f2 100644
--- a/action.yml
+++ b/action.yml
@@ -10,4 +10,7 @@ inputs:
     default: 'x64'
 runs:
   using: 'node12'
-  main: 'lib/setup-python.js'
+  main: 'dist/index.js'
+branding:
+  icon: 'play'
+  color: 'yellow'
\ No newline at end of file