diff --git a/custom/testing/golden-state-tree/os/amazon/pkgs/init.sls b/custom/testing/golden-state-tree/os/amazon/pkgs/init.sls
index 7e12bbb..965dd43 100644
--- a/custom/testing/golden-state-tree/os/amazon/pkgs/init.sls
+++ b/custom/testing/golden-state-tree/os/amazon/pkgs/init.sls
@@ -3,6 +3,7 @@ include:
   - pkgs.curl
   - pkgs.dmidecode
   - pkgs.dnsutils
+  - pkgs.docker
   - pkgs.gcc
   - pkgs.gpg
   - pkgs.libcurl
diff --git a/custom/testing/golden-state-tree/os/amazon2023/pkgs/init.sls b/custom/testing/golden-state-tree/os/amazon2023/pkgs/init.sls
index 131f56f..695d5da 100644
--- a/custom/testing/golden-state-tree/os/amazon2023/pkgs/init.sls
+++ b/custom/testing/golden-state-tree/os/amazon2023/pkgs/init.sls
@@ -3,6 +3,7 @@ include:
   - pkgs.curl
   - pkgs.dmidecode
   - pkgs.dnsutils
+  - pkgs.docker
   - pkgs.gcc
   - pkgs.gpg
   - pkgs.iproute2
diff --git a/custom/testing/golden-state-tree/os/photon/pkgs/init.sls b/custom/testing/golden-state-tree/os/photon/pkgs/init.sls
index 3bf743b..0cd3ba5 100644
--- a/custom/testing/golden-state-tree/os/photon/pkgs/init.sls
+++ b/custom/testing/golden-state-tree/os/photon/pkgs/init.sls
@@ -11,6 +11,7 @@ include:
   - pkgs.curl
   - pkgs.dmidecode
   - pkgs.dnsutils
+  - pkgs.docker
   - pkgs.gcc
   - pkgs.git
   - pkgs.gpg
diff --git a/custom/testing/golden-state-tree/pkgs/vault.sls b/custom/testing/golden-state-tree/pkgs/vault.sls
index a93c0c5..3ed1b29 100644
--- a/custom/testing/golden-state-tree/pkgs/vault.sls
+++ b/custom/testing/golden-state-tree/pkgs/vault.sls
@@ -23,7 +23,7 @@ vault-repo:
     - name: |
         dnf -y install dnf-plugins-core
         dnf config-manager --add-repo https://rpm.releases.hashicorp.com/fedora/hashicorp.repo
-  {%- elif grains['os'] in ('Amazon', 'Rocky') %}
+  {%- elif grains['os'] in ('Amazon') %}
   {#- Amazon must be addressed first because of the os_family logical check below #}
     - name: |
         yum install -y yum-utils