diff --git a/main.py b/main.py
index c083e30..10626e2 100644
--- a/main.py
+++ b/main.py
@@ -83,7 +83,7 @@ def main(
                 teacher_settings=dict(lm=teacher_lm),
             )
             compiled_baleen = optimizer.compile(
-                module_type(), trainset=datasets[question_type]["train"][:train_size]
+                module_type, trainset=datasets[question_type]["train"][:train_size]
             )
         # elif dspy_optimizer == "signature_optimizer": # Signature Optimizer is deprecated TODO: add a new one like GEPA
         #    optimizer = SignatureOptimizer(