diff --git a/trustgraph-flow/trustgraph/extract/kg/agent/extract.py b/trustgraph-flow/trustgraph/extract/kg/agent/extract.py
index b7ef9259..041264a4 100644
--- a/trustgraph-flow/trustgraph/extract/kg/agent/extract.py
+++ b/trustgraph-flow/trustgraph/extract/kg/agent/extract.py
@@ -322,6 +322,7 @@ class Processor(FlowProcessor):
 
         parser.add_argument(
             "--template-id",
+            dest="template_id",
             type=str,
             default=default_template_id,
             help="Template ID to use for agent extraction"
@@ -329,6 +330,7 @@ class Processor(FlowProcessor):
 
         parser.add_argument(
             '--config-type',
+            dest="config_type",
             default="prompt",
             help=f'Configuration key for prompts (default: prompt)',
         )