diff --git a/llama_toolchain/cli/stack/build.py b/llama_toolchain/cli/stack/build.py
index 9d1c1ad01..58d13b70a 100644
--- a/llama_toolchain/cli/stack/build.py
+++ b/llama_toolchain/cli/stack/build.py
@@ -51,7 +51,7 @@ class StackBuild(Subcommand):
 
         # save build.yaml spec for building same distribution again
         if build_config.image_type == ImageType.docker.value:
-            # TODO (xiyan): docker needs build file to be in the llama-stack repo dir
+            # docker needs build file to be in the llama-stack repo dir to be able to copy over to the image
             llama_toolchain_path = Path(os.path.relpath(__file__)).parent.parent.parent
             build_dir = (
                 llama_toolchain_path / "configs/distributions" / build_config.image_type