diff --git a/README.md b/README.md
index e8ab1425..ed12c89c 100644
--- a/README.md
+++ b/README.md
@@ -11,7 +11,9 @@ Machine Learning Collective Variables for Enhanced Sampling
 **PAPER** [![paper](https://img.shields.io/badge/JCP-10.1063/5.0156343-blue)](https://doi.org/10.1063/5.0156343)
 [![preprint](https://img.shields.io/badge/arXiv-2305.19980-lightblue)](https://arxiv.org/abs/2305.19980)
 
-The documentation is available here: https://mlcolvar.readthedocs.io
+The documentation is available at:
+- **stable** version: https://mlcolvar.readthedocs.io
+- **latest** version: https://mlcolvar.readthedocs.io/en/latest/
 
 ---
 
diff --git a/mlcolvar/data/datamodule.py b/mlcolvar/data/datamodule.py
index b0446d55..4323b211 100644
--- a/mlcolvar/data/datamodule.py
+++ b/mlcolvar/data/datamodule.py
@@ -41,8 +41,8 @@ class DictModule(lightning.LightningDataModule):
     test sets.
 
     The class can also merge multiple :class:`~mlcolvar.data.dataset.DictDataset`s
-    that have different keys (see example below). The datasets must all have the
-    same number of samples.
+    that have different keys (see example below). The datasets can have different sizes but must all have the
+    same number of batches.
 
     Examples
     --------