diff --git a/.azure/azure-ci-build-pipeline.yaml b/.azure/azure-ci-build-pipeline.yaml
index 2dffe61a..a59c5593 100644
--- a/.azure/azure-ci-build-pipeline.yaml
+++ b/.azure/azure-ci-build-pipeline.yaml
@@ -98,7 +98,7 @@ stages:
             displayName: 'Perform PrimAITE Setup'
 
           - script: |
-              pytest -v tests/unit_tests --cov=src/ -o junit_family=xunit2 --junitxml=junit/test-results.xml --cov-report xml:$(System.DefaultWorkingDirectory)/coverage.xml --cov-report html:$(System.DefaultWorkingDirectory)/htmlcov --cov-report term
+              pytest -v tests/ --cov=src/ -o junit_family=xunit2 --junitxml=junit/test-results.xml --cov-report xml:$(System.DefaultWorkingDirectory)/coverage.xml --cov-report html:$(System.DefaultWorkingDirectory)/htmlcov --cov-report term
             displayName: 'Run tests and code coverage'
 
           - task: PublishTestResults@2
@@ -115,7 +115,7 @@ stages:
             displayName: 'Publish coverage report'
             inputs:
               codeCoverageTool: Cobertura
-              summaryFileLocation: '$(System.DefaultWorkingDirectory)/**/coverage.xml'
+              summaryFileLocation: '$(System.DefaultWorkingDirectory)/coverage.xml'
               pathToSources: '$(System.DefaultWorkingDirectory)/src'
               # reportDirectory: '$(System.DefaultWorkingDirectory)/**/htmlcov/'
               # additionalCodeCoverageFiles: '$(System.DefaultWorkingDirectory)/**/htmlcov/*.*'