diff --git a/docs/shimmer_basics.md b/docs/shimmer_basics.md
index 1af9b629..a5d7281a 100644
--- a/docs/shimmer_basics.md
+++ b/docs/shimmer_basics.md
@@ -334,8 +334,8 @@ from shimmer import RepeatedDataset
 class GWDataModule(LightningDataModule):
     def __init__(
         self,
-        val_datasets: dict[frozenset[str], DomainDataset],
         train_datasets: dict[frozenset[str], DomainDataset],
+        val_datasets: dict[frozenset[str], DomainDataset],
         batch_size: int,
     ) -> None:
         super().__init__()