diff --git a/dust3r/datasets/megadepth.py b/dust3r/datasets/megadepth.py
index 8131498..7d87b4d 100644
--- a/dust3r/datasets/megadepth.py
+++ b/dust3r/datasets/megadepth.py
@@ -14,7 +14,7 @@
 
 
 class MegaDepth(BaseStereoViewDataset):
-    def __init__(self, *args, split, ROOT, **kwargs):
+    def __init__(self, *args, ROOT, **kwargs):
         self.ROOT = ROOT
         super().__init__(*args, **kwargs)
         self.loaded_data = self._load_data(self.split)