From 88260a8c0a9d0752f4d32119300d96702c4395e0 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Wed, 30 Jul 2025 19:41:57 -0700
Subject: [PATCH] fixes

---
 llama_stack/core/build.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/llama_stack/core/build.py b/llama_stack/core/build.py
index 2eb10a3d3..a6544435f 100644
--- a/llama_stack/core/build.py
+++ b/llama_stack/core/build.py
@@ -122,7 +122,7 @@ def build_image(
                 normal_deps.extend(api_spec.pip_packages)
 
     if build_config.image_type == LlamaStackImageType.CONTAINER.value:
-        script = str(importlib.resources.files("llama_stack") / "distribution/build_container.sh")
+        script = str(importlib.resources.files("llama_stack") / "core/build_container.sh")
         args = [
             script,
             "--template-or-config",
@@ -139,7 +139,7 @@ def build_image(
         if run_config is not None:
             args.extend(["--run-config", run_config])
     elif build_config.image_type == LlamaStackImageType.CONDA.value:
-        script = str(importlib.resources.files("llama_stack") / "distribution/build_conda_env.sh")
+        script = str(importlib.resources.files("llama_stack") / "core/build_conda_env.sh")
         args = [
             script,
             "--env-name",
@@ -150,7 +150,7 @@ def build_image(
             " ".join(normal_deps),
         ]
     elif build_config.image_type == LlamaStackImageType.VENV.value:
-        script = str(importlib.resources.files("llama_stack") / "distribution/build_venv.sh")
+        script = str(importlib.resources.files("llama_stack") / "core/build_venv.sh")
         args = [
             script,
             "--env-name",