diff --git a/action.yaml b/action.yaml
index c3b5d04..6edc3e4 100644
--- a/action.yaml
+++ b/action.yaml
@@ -71,12 +71,16 @@ inputs:
     description: 'OS version on which splunk will be spinned for scripted input'
     required: false
     default: 'latest'
+  test-browser:
+    description: 'Browser type for UI tests'
+    required: false
+    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:v4.0.0'
+  image: 'Dockerfile'
   args:
     - ${{ inputs.workflow-tmpl-name }}
     - ${{ inputs.workflow-template-ns }}
@@ -94,3 +98,4 @@ runs:
     - ${{ inputs.sc4s-docker-registry }}
     - ${{ inputs.os-name }}
     - ${{ inputs.os-version }}
+    - ${{ inputs.test-browser }}
diff --git a/entrypoint.sh b/entrypoint.sh
index 1ab8599..2d3dc1d 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -40,6 +40,7 @@ WORKFLOW_NAME=`argo submit -v -o json --from wftmpl/${1} -n ${2} -l workflows.ar
     -p sc4s-docker-registry=${14} \
     -p os-name=${15} \
     -p os-version=${16} \
+    -p test-browser=${17} \
     -l="${9},test-type=${6},splunk-version=${5}" | jq -r .metadata.name`
 
 echo "After argo submit $?"