diff --git a/train.py b/train.py
index c87d514..cf777b1 100644
--- a/train.py
+++ b/train.py
@@ -66,7 +66,7 @@ def train(
     dataset = DatasetLoader.load(
         train_type=train_type,
         load_path=cache_files,
-        max_len=parameter.config.m_len
+        max_len=parameter.config.m_len,
         **kwargs
     )