diff --git a/src/rechunk_data/__init__.pyi b/src/rechunk_data/__init__.pyi
index 5817e93c573aa8079f6121fb4d0db590e0070a2d..6eca2dd0f35ebdd75a1577926ddadc1531026934 100644
--- a/src/rechunk_data/__init__.pyi
+++ b/src/rechunk_data/__init__.pyi
@@ -11,7 +11,7 @@ def _save_dataset(
     dset: xr.Dataset, file_name: Path, encoding: Dict[str, Any], engine: str
 ) -> None: ...
 def _rechunk_dataset(dset: xr.Dataset) -> Tuple[xr.Dataset, Dict[str, Any]]: ...
-def rechunk_dataset(dset: xr.Dataset) -> xr.dataset: ...
+def rechunk_dataset(dset: xr.Dataset) -> xr.Dataset: ...
 def rechunk_netcdf_file(
     input_path: os.PathLike,
     output_path: Optional[os.PathLike] = ...,