diff --git a/mofa/generator.py b/mofa/generator.py
index 1b88cbae..4dc5d2f0 100644
--- a/mofa/generator.py
+++ b/mofa/generator.py
@@ -10,7 +10,7 @@
 import os
 
 def train_generator
-        config: str="config.yaml"
+        config: str="config.yaml",
         # starting_model: str | Path,
         examples: str|list[MOFRecord]="../argonne_gnn_gitlab/DiffLinker/data/geom/datasets",
         num_epochs: int=10