diff --git a/action.yaml b/action.yaml
index 5f58f13..08782b3 100644
--- a/action.yaml
+++ b/action.yaml
@@ -79,12 +79,16 @@ inputs:
     required: false
     description: 'Initial version of the TA for upgrade tests'
     default: ''
+  python-version:
+    description: 'Python version to use for 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:v5.1.0'
+  image: 'Dockerfile'
   args:
     - ${{ inputs.workflow-tmpl-name }}
     - ${{ inputs.workflow-template-ns }}
@@ -104,3 +108,4 @@ runs:
     - ${{ inputs.os-version }}
     - ${{ inputs.test-browser }}
     - ${{ inputs.ta-upgrade-version }}
+    - ${{ inputs.python-version }}
diff --git a/entrypoint.sh b/entrypoint.sh
index 99cbaa9..c07a4c3 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -42,6 +42,7 @@ WORKFLOW_NAME=`argo submit -v -o json --from wftmpl/${1} -n ${2} -l workflows.ar
     -p os-version=${16} \
     -p test-browser=${17} \
     -p ta-upgrade-version=${18} \
+    -p python-version=${19} \
     -l="${9},test-type=${6},splunk-version=${5}" | jq -r .metadata.name`
 
 echo "After argo submit $?"