diff --git a/llama_stack/cli/stack/_build.py b/llama_stack/cli/stack/_build.py
index d5a9173ee..65d37e9da 100644
--- a/llama_stack/cli/stack/_build.py
+++ b/llama_stack/cli/stack/_build.py
@@ -56,7 +56,7 @@ def available_templates_specs() -> Dict[str, BuildConfig]:
     return template_specs
 
 
-def run_stack_build_command(parser: argparse.ArgumentParser, args: argparse.Namespace) -> None:
+def run_stack_build_command(args: argparse.Namespace) -> None:
     if args.list_templates:
         return _run_template_list_cmd()
 
diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
index 1b673b889..729bd3ff1 100644
--- a/llama_stack/cli/stack/build.py
+++ b/llama_stack/cli/stack/build.py
@@ -75,4 +75,4 @@ environment is active, you must specify a name.
         # can be fast to load and reduces dependencies
         from ._build import run_stack_build_command
 
-        return run_stack_build_command(self.parser, args)
+        return run_stack_build_command(args)