diff --git a/action.yml b/action.yml
index 61c08a4..304e465 100644
--- a/action.yml
+++ b/action.yml
@@ -7,6 +7,7 @@ inputs:
     required: true
     default: ${{ github.token }}
 
+
 outputs:
   base_ref_branch:
     description: 'The target branch of a pull request'
@@ -18,10 +19,6 @@ outputs:
 runs:
   using: 'docker'
   image: 'Dockerfile'
-  args:
-    - ${{ github.base_ref }}
-    - ${{ github.head_ref }}
-    - ${{ github.ref }}
 branding:
   icon: git-branch
   color: white
diff --git a/entrypoint.sh b/entrypoint.sh
index 0ff7b61..8784ced 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -2,6 +2,6 @@
 
 set -e
 
-echo "::set-output name=base_ref_branch::${0/refs\/heads\//}"
-echo "::set-output name=head_ref_branch::${1/refs\/heads\//}"
-echo "::set-output name=ref_branch::${2/refs\/heads\//}"
+echo "::set-output name=base_ref_branch::${INPUT_BASE_REF/refs\/heads\//}"
+echo "::set-output name=head_ref_branch::${INPUT_HEAD_REF/refs\/heads\//}"
+echo "::set-output name=ref_branch::${INPUT_REF/refs\/heads\//}"