diff --git a/setup.cfg b/setup.cfg
index 37462b6..3054b5c 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -74,13 +74,19 @@ deps =
     pytest-cov == 4.0.0
 
 commands =
-    pytest --cov=cocotbext --cov=tests --cov-branch {posargs:-n auto --verbose}
+    pytest {posargs:-n auto --verbose}
     bash -c 'find . -type f -name "\.coverage" | xargs coverage combine --append'
     coverage report
 
 allowlist_externals =
     bash
 
+[coverage:run]
+branch = True
+source =
+    cocotbext.axi
+    tests
+
 # combine if paths are different
 [coverage:paths]
 source =