diff --git a/pyproject.toml b/pyproject.toml
index a4e125f5a31..900aaa58979 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -148,6 +148,7 @@ show_missing = true
 
 [tool.coverage.run]
 source_pkgs = ["torchgeo"]
+branch = true
 
 # https://mypy.readthedocs.io/en/stable/config_file.html
 [tool.mypy]