diff --git a/templates/ImageBuilder/amazon-linux-2023-agora-bastian.yaml b/templates/ImageBuilder/amazon-linux-2023-agora-bastian.yaml
index 0a335bb..881d8ce 100644
--- a/templates/ImageBuilder/amazon-linux-2023-agora-bastian.yaml
+++ b/templates/ImageBuilder/amazon-linux-2023-agora-bastian.yaml
@@ -9,6 +9,12 @@ Parameters:
     Description: The generated image version.
     Type: String
     Default: "0.0.0"
+  VolumeSize:
+    Description: The EBS volume size (in GB)
+    Type: Number
+    Default: 32
+    MinValue: 8
+    MaxValue: 500
 Resources:
   # By default, AWS Services do not have permission to perform actions on your instances. This grants
   # AWS Systems Manager (SSM) and EC2 Image Builder the necessary permissions to build an image.
@@ -67,6 +73,10 @@ Resources:
       Components:
         - ComponentArn:
             Fn::Sub: arn:${AWS::Partition}:imagebuilder:${AWS::Region}:aws:component/update-linux/x.x.x
+      BlockDeviceMappings:
+        - DeviceName: /dev/xvda
+          Ebs:
+            VolumeSize: !Ref VolumeSize
       # Make sure the latest version of the Amazon SSM Agent is installed on the image.
       AdditionalInstanceConfiguration:
         UserDataOverride: