diff --git a/main.py b/main.py
index 4871fe4..4edbb1b 100644
--- a/main.py
+++ b/main.py
@@ -20,7 +20,7 @@ def main():
     model.generate(
         n_target_bar=16,
         temperature=1.2,
-        topk=5
+        topk=5,
         output_path='./result/continuation.midi',
         prompt='./data/evaluation/000.midi')