diff --git a/llama_stack/cli/stack/build.py b/llama_stack/cli/stack/build.py
index da3253de2..8f609fbad 100644
--- a/llama_stack/cli/stack/build.py
+++ b/llama_stack/cli/stack/build.py
@@ -191,8 +191,6 @@ class StackBuild(Subcommand):
 
         from llama_stack.distribution.build import ImageType
 
-        from llama_stack.distribution.utils.serialize import EnumEncoder
-
         apis = list(build_config.distribution_spec.providers.keys())
         run_config = StackRunConfig(
             built_at=datetime.now(),
@@ -250,7 +248,7 @@ class StackBuild(Subcommand):
 
         import yaml
 
-        from llama_stack.distribution.build import build_image, ImageType
+        from llama_stack.distribution.build import build_image
         from llama_stack.distribution.utils.config_dirs import DISTRIBS_BASE_DIR
 
         # save build.yaml spec for building same distribution again
diff --git a/llama_stack/cli/stack/configure.py b/llama_stack/cli/stack/configure.py
index 1647b468b..c2f8b68c2 100644
--- a/llama_stack/cli/stack/configure.py
+++ b/llama_stack/cli/stack/configure.py
@@ -39,7 +39,7 @@ class StackConfigure(Subcommand):
     def _run_stack_configure_cmd(self, args: argparse.Namespace) -> None:
         self.parser.error(
             """
-            DEPRECATED! llama stack configure has been deprecated. 
+            DEPRECATED! llama stack configure has been deprecated.
             Please use llama stack run --config <path/to/run.yaml> instead. You may find example run.yaml in ./llamas_stack/templates
             """
         )