diff --git a/llama_stack/distribution/build_container.sh b/llama_stack/distribution/build_container.sh
index 9bb02d84b..cc2ef215e 100755
--- a/llama_stack/distribution/build_container.sh
+++ b/llama_stack/distribution/build_container.sh
@@ -262,7 +262,7 @@ fi
 # Add other require item commands genearic to all containers
 add_to_container << EOF
 
-RUN chmod -R g+rw /app /.llama
+RUN mkdir -p /.llama && chmod -R g+rw /app /.llama
 EOF
 
 printf "Containerfile created successfully in %s/Containerfile\n\n" "$TEMP_DIR"