diff --git a/.azure/azure-ci-build-pipeline.yaml b/.azure/azure-ci-build-pipeline.yaml
index 3e2237d3..77cae6fc 100644
--- a/.azure/azure-ci-build-pipeline.yaml
+++ b/.azure/azure-ci-build-pipeline.yaml
@@ -106,7 +106,7 @@ stages:
               version: '2.1.x'
 
           - script: |
-              coverage run -m pytest -v -o junit_family=xunit2 --junitxml=junit/test-results.xml
+              coverage run -m --source=primaite pytest -v -o junit_family=xunit2 --junitxml=junit/test-results.xml
               coverage xml -o coverage.xml -i
               coverage html -d htmlcov -i
             displayName: 'Run tests and code coverage'