diff --git a/.azure/azure-ci-build-pipeline.yaml b/.azure/azure-ci-build-pipeline.yaml
index 5759e70e..36704ac0 100644
--- a/.azure/azure-ci-build-pipeline.yaml
+++ b/.azure/azure-ci-build-pipeline.yaml
@@ -104,7 +104,9 @@ stages:
               version: '2.1.x'
 
           - script: |
-              pytest -v --cov=$(System.DefaultWorkingDirectory)/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
+              coverage run -m pytest tests/unit_tests
+              coverage xml -o coverage.xml -i
+              coverage html -d htmlcov -i
             displayName: 'Run tests and code coverage'
 
           - task: PublishTestResults@2
@@ -117,6 +119,12 @@ stages:
           - publish: $(System.DefaultWorkingDirectory)/htmlcov/
             artifact: coverage_report
 
+          - task: PublishCodeCoverageResults@1
+            inputs:
+              codeCoverageTool: Cobertura
+              summaryFileLocation: '$(System.DefaultWorkingDirectory)/**/coverage.xml'
+              reportDirectory: '$(System.DefaultWorkingDirectory)/**/htmlcov'
+
           # - task: PublishCodeCoverageResults@2
           #   displayName: 'Publish coverage report'
           #   inputs: