From ba15bf315bced8cac3e8f9c90113abeceb253a6a Mon Sep 17 00:00:00 2001
From: Hope Aitchison <hope.aitchison@digital.justice.gov.uk>
Date: Fri, 20 Oct 2023 17:50:36 +0100
Subject: [PATCH] updating tasks

---
 .../ssm-documents/ami-build.yaml                     | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/terraform/environments/corporate-staff-rostering/ssm-documents/ami-build.yaml b/terraform/environments/corporate-staff-rostering/ssm-documents/ami-build.yaml
index 7671c1039f0..dc1cfcd56d7 100644
--- a/terraform/environments/corporate-staff-rostering/ssm-documents/ami-build.yaml
+++ b/terraform/environments/corporate-staff-rostering/ssm-documents/ami-build.yaml
@@ -17,11 +17,13 @@ parameters:
 
 mainSteps:
   - name: CreateSnapshot
-    action: aws:createSnapshot
+    action: aws:runDocument
     inputs: 
-      InstanceId: "{{ InstanceId }}"
-      VolumeId: "{{ VolumeId }}"
-      description: "{{ SnapshotName }}"
+      documentType: SSMDocument
+      documentPath: AWS-CreateSnapshot
+      documentParameters:
+        VolumeId: "{{ VolumeId }}"
+        description: "{{ SnapshotName }}"
     onFailure: Abort
 
   - name: getRootVolumeSpace
@@ -162,7 +164,7 @@ mainSteps:
         - |  
           Write-Output "Image ID: $ImageId"
 
-  - name: waitForAMI
+  - name: waitForAMI # update this
     action: aws:waitForImages
     inputs:
       imageIds: ["{{ createImage.ImageId }}"]