diff --git a/agent/__init__.py b/agent/__init__.py
index 98962fa..4c37f31 100644
--- a/agent/__init__.py
+++ b/agent/__init__.py
@@ -1,3 +1,3 @@
-from .modules import SignatureGenerator
+from .modules import signature_generator
 
-__all__ = ["SignatureGenerator"]
+__all__ = ["signature_generator"]
diff --git a/agent/modules.py b/agent/modules.py
index d4e10d1..ca08bca 100644
--- a/agent/modules.py
+++ b/agent/modules.py
@@ -238,3 +238,5 @@ class SignatureGenerator(dspy.Module):
             )
 
         return imports
+
+signature_generator = SignatureGenerator()
\ No newline at end of file
diff --git a/main.py b/main.py
index bb5676f..2135967 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 from modaic import PrecompiledAgent, PrecompiledConfig
-from agent.modules import SignatureGenerator
+from agent import signature_generator
 import dspy
 
 
@@ -16,7 +16,7 @@ class PromptToSignatureAgent(PrecompiledAgent):
 
     def __init__(self, config: PromptToSignatureConfig, **kwargs):
         super().__init__(config, **kwargs)
-        self.signature_generator = SignatureGenerator()
+        self.signature_generator = signature_generator
         self.signature_refiner = dspy.Refine(
             module=self.signature_generator,
             N=config.max_attempts_to_refine,