diff --git a/torchgeo/datamodules/utils.py b/torchgeo/datamodules/utils.py
index 51a07bcfca..12641a09d6 100644
--- a/torchgeo/datamodules/utils.py
+++ b/torchgeo/datamodules/utils.py
@@ -171,7 +171,7 @@ def group_shuffle_split(
     return train_idxs, test_idxs
 
 
-def get_prefixed_kwargs(prefix: str, **kwargs: dict[str, Any]) -> dict[str, Any]:
+def get_prefixed_kwargs(prefix: str, **kwargs: Any) -> Any:
     """Get kwargs with a specific prefix.
 
     Args: