diff --git a/ahcore/data/dataset.py b/ahcore/data/dataset.py
index 050326b..da10b97 100644
--- a/ahcore/data/dataset.py
+++ b/ahcore/data/dataset.py
@@ -86,10 +86,12 @@ def __len__(self) -> int:
         return self.cumulative_sizes[-1]
 
     @overload
-    def __getitem__(self, index: int) -> DlupDatasetSample: ...
+    def __getitem__(self, index: int) -> DlupDatasetSample:
+        ...
 
     @overload
-    def __getitem__(self, index: slice) -> list[DlupDatasetSample]: ...
+    def __getitem__(self, index: slice) -> list[DlupDatasetSample]:
+        ...
 
     def __getitem__(self, index: Union[int, slice]) -> DlupDatasetSample | list[DlupDatasetSample]:
         """Returns the sample at the given index."""
@@ -110,7 +112,6 @@ def __init__(
         self,
         data_description: DataDescription,
         pre_transform: Callable[[bool], Callable[[DlupDatasetSample], DlupDatasetSample]],
-        database_uri: str | None = None,
         batch_size: int = 32,  # noqa,pylint: disable=unused-argument
         validate_batch_size: int | None = None,  # noqa,pylint: disable=unused-argument
         num_workers: int = 16,
@@ -128,8 +129,6 @@ def __init__(
             A pre-transform is a callable which is directly applied to the output of the dataset before collation in
             the dataloader. The transforms typically convert the image in the output to a tensor, convert the
             `WsiAnnotations` to a mask or similar.
-        database_uri : str, optional
-            The URI to the database. If not provided, the URI from the data description will be used.
         batch_size : int
             The batch size of the data loader.
         validate_batch_size : int, optional
@@ -161,9 +160,7 @@ def __init__(
         # Data settings
         self.data_description: DataDescription = data_description
 
-        self._data_manager = DataManager(
-            database_uri=data_description.manifest_database_uri if not database_uri else database_uri
-        )
+        self._data_manager = DataManager(database_uri=data_description.manifest_database_uri)
 
         self._batch_size = self.hparams.batch_size  # type: ignore
         self._validate_batch_size = self.hparams.validate_batch_size  # type: ignore