diff --git a/action.yaml b/action.yaml
index 71a0c7f..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.1'
+  image: 'Dockerfile'
   args:
     - ${{ inputs.workflow-tmpl-name }}
     - ${{ inputs.workflow-template-ns }}
diff --git a/entrypoint.sh b/entrypoint.sh
index 2d3dc1d..7818874 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -47,4 +47,4 @@ echo "After argo submit $?"
 echo 'WORKFLOW_NAME:' ${WORKFLOW_NAME}
 echo "workflow-name=$(echo $WORKFLOW_NAME)" >> $GITHUB_OUTPUT
 
-argo logs --follow ${WORKFLOW_NAME} -n ${2}
+argo logs --follow ${WORKFLOW_NAME} -n ${2} || echo "... There was an error fetching the logs. The workflow is still in progress, and the tests are still running. The results will be available in the Test Report step. Please wait for the workflow to complete. ..."