diff --git a/scripts/run_agent_server.py b/scripts/run_agent_server.py
index 4b7bed8..b8c84a0 100644
--- a/scripts/run_agent_server.py
+++ b/scripts/run_agent_server.py
@@ -2,11 +2,11 @@ import tyro
 import asyncio
 
 from lang_agent.pipeline import Pipeline, PipelineConfig
-from lang_agent.config import load_config
+from lang_agent.config import load_tyro_conf
 
 def main(conf:PipelineConfig):
     if conf.config_f is not None:
-        conf = load_config(conf.config_f)
+        conf = load_tyro_conf(conf.config_f)
     
     pipeline:Pipeline = conf.setup()
     asyncio.run(pipeline.start_server())