diff --git a/Dockerfile b/Dockerfile
index e2b3b78..513f5b8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ghcr.io/splunk/workflow-engine-base:4.0.0
+FROM ghcr.io/splunk/workflow-engine-base:4.1.0
 
 COPY entrypoint.sh /entrypoint.sh
 
diff --git a/action.yaml b/action.yaml
index 3a719c3..6edc3e4 100644
--- a/action.yaml
+++ b/action.yaml
@@ -80,7 +80,7 @@ outputs:
     description: 'Name of workflow triggered'
 runs:
   using: 'docker'
-  image: 'docker://ghcr.io/splunk/wfe-test-runner-action/wfe-test-runner-action:v5.0.0'
+  image: 'Dockerfile'
   args:
     - ${{ inputs.workflow-tmpl-name }}
     - ${{ inputs.workflow-template-ns }}