diff --git a/src/ptbench/data/montgomery_shenzhen/datamodule.py b/src/ptbench/data/montgomery_shenzhen/datamodule.py
index 35ba9b9a04cace630fcced8aa15a7333ec4d7387..0335a2e691ed2fcf10378e501d7715aa474c4475 100644
--- a/src/ptbench/data/montgomery_shenzhen/datamodule.py
+++ b/src/ptbench/data/montgomery_shenzhen/datamodule.py
@@ -15,22 +15,22 @@ class DataModule(ConcatDataModule):
     def __init__(self, split_filename: str):
         montgomery_loader = MontgomeryLoader()
         montgomery_split = make_montgomery_split("default.json")
-        shenzen_loader = ShenzhenLoader()
-        shenzen_split = make_shenzhen_split("default.json")
+        shenzhen_loader = ShenzhenLoader()
+        shenzhen_split = make_shenzhen_split("default.json")
 
         super().__init__(
             splits={
                 "train": [
                     (montgomery_split["train"], montgomery_loader),
-                    (shenzen_split["train"], shenzen_loader),
+                    (shenzhen_split["train"], shenzhen_loader),
                 ],
                 "validation": [
                     (montgomery_split["validation"], montgomery_loader),
-                    (shenzen_split["validation"], shenzen_loader),
+                    (shenzhen_split["validation"], shenzhen_loader),
                 ],
                 "test": [
                     (montgomery_split["test"], montgomery_loader),
-                    (shenzen_split["test"], shenzen_loader),
+                    (shenzhen_split["test"], shenzhen_loader),
                 ],
             }
         )