diff --git a/.azure/azure-ci-build-pipeline.yaml b/.azure/azure-ci-build-pipeline.yaml
index 8d418ba5..01111290 100644
--- a/.azure/azure-ci-build-pipeline.yaml
+++ b/.azure/azure-ci-build-pipeline.yaml
@@ -123,13 +123,12 @@ stages:
           # Run notebooks
           - script: |
               pytest --nbmake -n=auto src/primaite/notebooks --junit-xml=./notebook-tests/notebooks.xml
-              $notebooks_exit_code = $LASTEXITCODE
+              set notebooks_exit_code=%ERRORLEVEL%
               pytest --nbmake -n=auto src/primaite/simulator/_package_data --junit-xml=./notebook-tests/package-notebooks.xml
-              $package_notebooks_exit_code = $LASTEXITCODE
-              # Fail step if either of these do not have exit code 0
-              if ($notebooks_exit_code -ne 0 -or $package_notebooks_exit_code -ne 0) {
-                exit 1
-              }
+              set package_notebooks_exit_code=%ERRORLEVEL%
+              rem Fail step if either of these do not have exit code 0
+              if %notebooks_exit_code% NEQ 0 exit /b 1
+              if %package_notebooks_exit_code% NEQ 0 exit /b 1
             displayName: 'Run notebooks on Windows'
             condition: eq(variables['Agent.OS'], 'Windows_NT')