diff --git a/README.md b/README.md
index 540c2a6..19f6046 100644
--- a/README.md
+++ b/README.md
@@ -223,7 +223,7 @@ for example.
 | <a name="input_cloudwatch_logs"></a> [cloudwatch\_logs](#input\_cloudwatch\_logs) | CloudWatch logs configuration for the containers of this service. CloudWatch logs will be used as the default log configuration if Firelens is disabled and for the fluentbit and otel containers. | <pre>object({<br>    enabled           = optional(bool, true)<br>    name              = optional(string, "")<br>    retention_in_days = optional(number, 7)<br>  })</pre> | `{}` | no |
 | <a name="input_cluster_id"></a> [cluster\_id](#input\_cluster\_id) | The ECS cluster id that should run this service | `string` | n/a | yes |
 | <a name="input_code_build_environment_compute_type"></a> [code\_build\_environment\_compute\_type](#input\_code\_build\_environment\_compute\_type) | Information about the compute resources the CodeBuild stage of the deployment pipeline will use. | `string` | `"BUILD_LAMBDA_1GB"` | no |
-| <a name="input_code_build_environment_image"></a> [code\_build\_environment\_image](#input\_code\_build\_environment\_image) | Docker image to use for the CodeBuild stage of the deployment pipeline. | `string` | `"aws/codebuild/amazonlinux-aarch64-lambda-standard:python3.12"` | no |
+| <a name="input_code_build_environment_image"></a> [code\_build\_environment\_image](#input\_code\_build\_environment\_image) | Docker image to use for the CodeBuild stage of the deployment pipeline. The image needs to include python. | `string` | `"aws/codebuild/amazonlinux-aarch64-lambda-standard:python3.12"` | no |
 | <a name="input_code_build_environment_type"></a> [code\_build\_environment\_type](#input\_code\_build\_environment\_type) | Type of build environment for the CodeBuild stage of the deployment pipeline. | `string` | `"ARM_LAMBDA_CONTAINER"` | no |
 | <a name="input_code_build_log_retention_in_days"></a> [code\_build\_log\_retention\_in\_days](#input\_code\_build\_log\_retention\_in\_days) | Log retention in days of the CodeBuild CloudWatch log group. | `number` | `7` | no |
 | <a name="input_code_build_role_name"></a> [code\_build\_role\_name](#input\_code\_build\_role\_name) | Use an existing role for codebuild permissions that can be reused for multiple services. Otherwise a separate role for this service will be created. | `string` | `""` | no |
diff --git a/modules/deployment/code_build.tf b/modules/deployment/code_build.tf
index 3411cfb..c838da4 100644
--- a/modules/deployment/code_build.tf
+++ b/modules/deployment/code_build.tf
@@ -49,9 +49,6 @@ resource "aws_codebuild_project" "this" {
 version: 0.2
 
 phases:
-  install:
-    runtime-versions:
-      python: 3.12
   build:
     commands:
       - |
diff --git a/modules/deployment/variables.tf b/modules/deployment/variables.tf
index 5aeaa25..f43edcd 100644
--- a/modules/deployment/variables.tf
+++ b/modules/deployment/variables.tf
@@ -53,7 +53,7 @@ variable "code_build_environment_compute_type" {
 }
 
 variable "code_build_environment_image" {
-  description = "Docker image to use for the CodeBuild stage of the deployment pipeline."
+  description = "Docker image to use for the CodeBuild stage of the deployment pipeline. The image needs to include python."
   default     = "aws/codebuild/amazonlinux-aarch64-lambda-standard:python3.12"
   type        = string
 }
diff --git a/variables.tf b/variables.tf
index ced0d30..945233e 100644
--- a/variables.tf
+++ b/variables.tf
@@ -156,7 +156,7 @@ variable "code_build_environment_compute_type" {
 }
 
 variable "code_build_environment_image" {
-  description = "Docker image to use for the CodeBuild stage of the deployment pipeline."
+  description = "Docker image to use for the CodeBuild stage of the deployment pipeline. The image needs to include python."
   default     = "aws/codebuild/amazonlinux-aarch64-lambda-standard:python3.12"
   type        = string
 }