diff --git a/llama_toolchain/cli/api/start.py b/llama_toolchain/cli/api/start.py
index 539679561..a162d8dcb 100644
--- a/llama_toolchain/cli/api/start.py
+++ b/llama_toolchain/cli/api/start.py
@@ -62,13 +62,13 @@ class ApiStart(Subcommand):
         if config.docker_image:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_container.sh",
+                "core/start_container.sh",
             )
             run_args = [script, config.docker_image]
         else:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_conda_env.sh",
+                "core/start_conda_env.sh",
             )
             run_args = [
                 script,
diff --git a/llama_toolchain/cli/stack/start.py b/llama_toolchain/cli/stack/start.py
index 6b8981128..433d5b8dc 100644
--- a/llama_toolchain/cli/stack/start.py
+++ b/llama_toolchain/cli/stack/start.py
@@ -70,13 +70,13 @@ class StackStart(Subcommand):
         if config.docker_image:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_container.sh",
+                "core/start_container.sh",
             )
             run_args = [script, config.docker_image]
         else:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_conda_env.sh",
+                "core/start_conda_env.sh",
             )
             run_args = [
                 script,
diff --git a/llama_toolchain/core/package.py b/llama_toolchain/core/package.py
index 9cb58569d..7ef68bcc9 100644
--- a/llama_toolchain/core/package.py
+++ b/llama_toolchain/core/package.py
@@ -147,7 +147,7 @@ def build_package(
 
     if build_type == BuildType.container:
         script = pkg_resources.resource_filename(
-            "llama_toolchain", "distribution/build_container.sh"
+            "llama_toolchain", "core/build_container.sh"
         )
         args = [
             script,
@@ -158,7 +158,7 @@ def build_package(
         ]
     else:
         script = pkg_resources.resource_filename(
-            "llama_toolchain", "distribution/build_conda_env.sh"
+            "llama_toolchain", "core/build_conda_env.sh"
         )
         args = [
             script,