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 }}"]