diff --git a/action.yaml b/action.yaml
index 1f49462..6bdc2d8 100644
--- a/action.yaml
+++ b/action.yaml
@@ -75,12 +75,20 @@ inputs:
     description: 'Browser type for UI tests'
     required: false
     default: ''
+  app-id:
+    required: false
+    description: 'APP ID'
+    default: ''
+  ta-upgrade-version:
+    required: false
+    description: 'upgrade version of TA'
+    default: ''
 outputs:
   workflow-name: # id of output
     description: 'Name of workflow triggered'
 runs:
   using: 'docker'
-  image: 'docker://ghcr.io/splunk/wfe-test-runner-action/wfe-test-runner-action:v5.0.2'
+  image: 'Dockerfile'
   args:
     - ${{ inputs.workflow-tmpl-name }}
     - ${{ inputs.workflow-template-ns }}
@@ -99,3 +107,5 @@ runs:
     - ${{ inputs.os-name }}
     - ${{ inputs.os-version }}
     - ${{ inputs.test-browser }}
+    - ${{ inputs.app-id }}
+    - ${{ inputs.ta-upgrade-version }}
\ No newline at end of file
diff --git a/entrypoint.sh b/entrypoint.sh
index 7818874..82da441 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -41,6 +41,8 @@ WORKFLOW_NAME=`argo submit -v -o json --from wftmpl/${1} -n ${2} -l workflows.ar
     -p os-name=${15} \
     -p os-version=${16} \
     -p test-browser=${17} \
+    -p app-id=${18} \
+    -p ta-upgrade-version=${19} \
     -l="${9},test-type=${6},splunk-version=${5}" | jq -r .metadata.name`
 
 echo "After argo submit $?"