diff --git a/pyproject.toml b/pyproject.toml
index 5b50f776609..c91c6092049 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -316,6 +316,9 @@ split-on-trailing-comma = false
 [tool.ruff.lint.pydocstyle]
 convention = "google"
 
+[tool.coverage.run]
+branch = true
+
 # https://setuptools.pypa.io/en/latest/userguide/pyproject_config.html
 [tool.setuptools.dynamic]
 version = {attr = "torchgeo.__version__"}
@@ -324,4 +327,4 @@ version = {attr = "torchgeo.__version__"}
 torchgeo = ["py.typed"]
 
 [tool.setuptools.packages.find]
-include = ["torchgeo*"]
+include = ["torchgeo*"]
\ No newline at end of file