diff --git a/llama_toolchain/cli/stack/build.py b/llama_toolchain/cli/stack/build.py
index 4de578836..4eb83b6cc 100644
--- a/llama_toolchain/cli/stack/build.py
+++ b/llama_toolchain/cli/stack/build.py
@@ -184,7 +184,7 @@ class StackBuild(Subcommand):
 
         build_config = BuildConfig(
             name=args.name,
-            distribution_type=args.distribution,
+            distribution=args.distribution,
             package_type=args.package_type,
             api_providers=args.api_providers,
         )