diff --git a/.github/actions/azureml-test/action.yml b/.github/actions/azureml-test/action.yml
index 1381aae075..da4e1a403b 100644
--- a/.github/actions/azureml-test/action.yml
+++ b/.github/actions/azureml-test/action.yml
@@ -123,7 +123,7 @@ runs:
       id: exit_status
       run: echo ::set-output name=code::$(cat ${{inputs.PYTEST_EXIT_CODE}})
     - name: Check Success/Failure
-      if: ${{ steps.execute_tests.outputs.code != 0 }}
+      if: ${{ steps.exit_status.outputs.code != 0 }}
       uses: actions/github-script@v3
       with:
         script: |