diff --git a/mmp/a4/dataset.py b/mmp/a4/dataset.py
index d642866..5d466ca 100644
--- a/mmp/a4/dataset.py
+++ b/mmp/a4/dataset.py
@@ -90,7 +90,7 @@ def get_dataloader(
     num_workers: int,
     anchor_grid: np.ndarray,
     is_test: bool,
-    is_train: bool,
+    is_train: bool = False,
 ) -> DataLoader:
     dataset = MMP_Dataset(
         path_to_data=path_to_data,