diff --git a/main.py b/main.py
index ad01e2b..bb5676f 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 from modaic import PrecompiledAgent, PrecompiledConfig
-from agent import SignatureGenerator
+from agent.modules import SignatureGenerator
 import dspy
 
 
@@ -168,9 +168,9 @@ CR_PROMPT = """ You are Charlotte, an advanced knowledge graph connection reason
 
 def main():
     # try refine
-    refined_result = agent(
-        prompt=CR_PROMPT,
-    )
+    #refined_result = agent(
+    #    prompt=CR_PROMPT,
+    #)
 
     agent.push_to_hub("fadeleke/prompt-to-signature", with_code=True)