diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
index d6a95c568..d8ace28bc 100644
--- a/llama_stack/cli/stack/build.py
+++ b/llama_stack/cli/stack/build.py
@@ -34,6 +34,7 @@ class StackBuild(Subcommand):
             "config",
             type=str,
             default=None,
+            nargs="*",
             help="Path to a config file to use for the build. You may find example configs in llama_stack/distribution/example_configs. If not defined, you will be prompted for entering wizard",
         )
 
diff --git a/llama_stack/cli/stack/configure.py b/llama_stack/cli/stack/configure.py
index e8b9ab205..364345669 100644
--- a/llama_stack/cli/stack/configure.py
+++ b/llama_stack/cli/stack/configure.py
@@ -135,6 +135,11 @@ class StackConfigure(Subcommand):
             f.write(yaml.dump(to_write, sort_keys=False))
 
         cprint(
-            f"> YAML configuration has been written to {run_config_file}. You can now run `llama stack run {run_config_file}`",
+            f"> YAML configuration has been written to {run_config_file}.",
             color="blue",
         )
+
+        cprint(
+            f"You can now run `llama stack run {image_name} --port PORT` or `llama stack run {run_config_file} --port PORT`",
+            color="green",
+        )