diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
index d2c1bf324..f35c6ceab 100644
--- a/llama_stack/cli/stack/build.py
+++ b/llama_stack/cli/stack/build.py
@@ -118,8 +118,14 @@ class StackBuild(Subcommand):
             f"Build spec configuration saved at {str(build_file_path)}",
             color="blue",
         )
+
+        configure_name = (
+            build_config.name
+            if build_config.image_type == "conda"
+            else (f"llamastack-{build_config.name}")
+        )
         cprint(
-            f"You may now run `llama stack configure {build_config.name}` or `llama stack configure {str(build_file_path)}`",
+            f"You may now run `llama stack configure {configure_name}` or `llama stack configure {str(build_file_path)}`",
             color="green",
         )
 
diff --git a/llama_stack/distribution/configure_container.sh b/llama_stack/distribution/configure_container.sh
index 49ef439a9..68009f8b1 100755
--- a/llama_stack/distribution/configure_container.sh
+++ b/llama_stack/distribution/configure_container.sh
@@ -28,4 +28,4 @@ set -x
 podman run -it \
   -v $host_build_dir:$container_build_dir \
   $docker_image \
-  llama stack configure --config ./llamastack-build.yaml --output-dir $container_build_dir
+  llama stack configure ./llamastack-build.yaml --output-dir $container_build_dir