diff --git a/.atmos/cache.yaml b/.atmos/cache.yaml
new file mode 100644
index 0000000..af84c06
--- /dev/null
+++ b/.atmos/cache.yaml
@@ -0,0 +1,3 @@
+installation_id: f67dcc7a-7060-475c-8933-19f2620f819b
+last_checked: 0
+telemetry_disclosure_shown: false
diff --git a/Makefile b/Makefile
index fbf0ae6..fcee3dc 100644
--- a/Makefile
+++ b/Makefile
@@ -34,3 +34,9 @@ test:
 
 cleantest:
 	cd test && docker compose down
+
+readme:
+	@atmos readme
+
+readme/build:
+	@atmos readme
diff --git a/README.md b/README.md
index bd5bd13..a247c90 100644
--- a/README.md
+++ b/README.md
@@ -2,8 +2,11 @@
 
 <!-- markdownlint-disable -->
 <a href="https://cpco.io/homepage"><img src="https://github.com/cloudposse/bastion/blob/main/.github/banner.png?raw=true" alt="Project Banner"/></a><br/>
-    <p align="right">
-<a href="https://github.com/cloudposse/bastion/releases/latest"><img src="https://img.shields.io/github/release/cloudposse/bastion.svg" alt="Latest Release"/></a><a href="https://github.com/cloudposse/bastion/actions/workflows/integration-tests.yml"><img src="https://github.com/cloudposse/bastion/actions/workflows/integration-tests.yml/badge.svg" alt="Build & Test Status"/></a><a href="https://slack.cloudposse.com"><img src="https://slack.cloudposse.com/badge.svg" alt="Slack Community"/></a></p>
+
+
+<p align="right"><a href="https://github.com/cloudposse/bastion/releases/latest"><img src="https://img.shields.io/github/release/cloudposse/bastion.svg" alt="Latest Release"/></a><a href="https://github.com/cloudposse/bastion/actions/workflows/integration-tests.yml"><img src="https://github.com/cloudposse/bastion/actions/workflows/integration-tests.yml/badge.svg" alt="Build & Test Status"/></a><a href="https://slack.cloudposse.com"><img src="https://slack.cloudposse.com/badge.svg" alt="Slack Community"/></a>
+
+</p>
 <!-- markdownlint-restore -->
 
 <!--
@@ -15,8 +18,8 @@
   **
   ** This file was automatically generated by the `cloudposse/build-harness`.
   ** 1) Make all changes to `README.yaml`
-  ** 2) Run `make init` (you only need to do this once)
-  ** 3) Run`make readme` to rebuild this file.
+  ** 2) Install [atmos](https://atmos.tools/install/) (you only need to do this once)
+  ** 3) Run`atmos readme` to rebuild this file.
   **
   ** (We maintain HUNDREDS of open source projects. This is how we maintain our sanity.)
   **
@@ -207,6 +210,10 @@ The first time you connect, you'll be asked to setup your MFA device. Subsequent
 
 
 
+
+
+
+
 ## ✨ Contributing
 
 This project is under active development, and we encourage contributions from our community.
diff --git a/atmos.yaml b/atmos.yaml
new file mode 100644
index 0000000..dfa6aca
--- /dev/null
+++ b/atmos.yaml
@@ -0,0 +1,2 @@
+import:
+  - https://raw.githubusercontent.com/cloudposse/.github/refs/heads/main/.github/atmos/github-action.yaml
diff --git a/test/docker-compose.yml b/test/docker-compose.yml
index fc32f4c..0621f89 100644
--- a/test/docker-compose.yml
+++ b/test/docker-compose.yml
@@ -13,9 +13,9 @@ services:
       MFA_PROVIDER: "google-authenticator"
     healthcheck:
       test: ["CMD", "nc", "-z", "localhost", "22"]
-      interval: 10s
-      timeout: 5s
-      retries: 3
+      interval: 20s
+      timeout: 10s
+      retries: 10
 
   test:
     build: "."
diff --git a/test/fixtures/client_scripts/google_auth_test.sh b/test/fixtures/client_scripts/google_auth_test.sh
index 2b20f7c..d242fc3 100755
--- a/test/fixtures/client_scripts/google_auth_test.sh
+++ b/test/fixtures/client_scripts/google_auth_test.sh
@@ -1,10 +1,28 @@
 #!/bin/sh
 ping -c 1 -w 5 bastion > /dev/null
 
+# Wait for SSH to be available on port 22
+max_attempts=10
+attempt=1
+while [ $attempt -le $max_attempts ]; do
+	if nc -z -w 2 bastion 22 2>/dev/null; then
+		break
+	fi
+	if [ $attempt -eq $max_attempts ]; then
+		echo "Error: SSH port 22 not available on bastion after $max_attempts attempts" >&2
+		exit 1
+	fi
+	attempt=$((attempt + 1))
+	sleep 1
+done
+
 # Add -vv for debugging.
 sshpass \
 	-P 'Verification code:' \
 	-f ./code \
 	ssh bastion@bastion \
+	-i /root/.ssh/id_rsa \
 	-o StrictHostKeyChecking=no \
+	-o IdentitiesOnly=yes \
 	-- echo 'this is a test.'
+
diff --git a/test/fixtures/client_scripts/sshrc_kill_test.sh b/test/fixtures/client_scripts/sshrc_kill_test.sh
index c868351..d507123 100755
--- a/test/fixtures/client_scripts/sshrc_kill_test.sh
+++ b/test/fixtures/client_scripts/sshrc_kill_test.sh
@@ -1,5 +1,21 @@
 #!/bin/sh
-ping -c 1 -w 5 bastion
+ping -c 1 -w 5 bastion > /dev/null
+
+# Wait for SSH to be available on port 22
+max_attempts=10
+attempt=1
+while [ $attempt -le $max_attempts ]; do
+	if nc -z -w 2 bastion 22 2>/dev/null; then
+		break
+	fi
+	if [ $attempt -eq $max_attempts ]; then
+		echo "Error: SSH port 22 not available on bastion after $max_attempts attempts" >&2
+		exit 1
+	fi
+	attempt=$((attempt + 1))
+	sleep 1
+done
+
 
 # Add -vv for debugging.
 sshpass \
diff --git a/test/test.sh b/test/test.sh
index 14d4af2..58e3da3 100755
--- a/test/test.sh
+++ b/test/test.sh
@@ -7,12 +7,14 @@ green=`tput setaf 2`
 reset=`tput sgr0`
 
 # Generating temp keys
+echo "Generating test ssh keys..."
 rm -rf fixtures/auth/ida_rsa*
 ssh-keygen -q -f fixtures/auth/ida_rsa -N ""
 chmod 600 fixtures/auth/ida_rsa
+echo "ssh keys generated."
 
 docker compose down
-docker compose up -d --build bastion
+docker compose up --wait -d --build bastion
 docker compose build test
 
 # wait until bastion is up
@@ -22,7 +24,6 @@ echo "Bastion sshd service started."
 
 docker compose exec bastion /scripts/setup.sh
 
-
 # greping for the first line of the left alignment square in the generated QR Code
 docker compose exec bastion /scripts/google_auth_qr_code_generator_test.sh |grep -F "                                                                                          " > /dev/null
 
@@ -36,12 +37,17 @@ else
 fi
 
 
-docker compose run --remove-orphans test /scripts/google_auth_test.sh
+# Ensure bastion is running before running test
+docker compose up -d --no-recreate bastion
+
+docker compose run --no-deps --rm test /scripts/google_auth_test.sh
 
 retVal=$?
 
 if [ $retVal -ne 0 ]; then
   echo "${red}* Google Authenticator/SSH Test Failed${reset}"
+  echo "Logs from bastion (executed before the failure):"
+  docker logs test-bastion-1
   exit $retVal
 else
   echo "${green}* Google Authenticator/SSH Test Succeeded${reset}"
@@ -70,7 +76,10 @@ else
   echo "${green}* Slack API Connection Test Succeeded${reset}"
 fi
 
-export SSHRC_KILL_OUTPUT=`docker compose run --remove-orphans test /scripts/sshrc_kill_test.sh`
+# Ensure bastion is running before running test
+docker compose up -d --no-recreate bastion
+
+export SSHRC_KILL_OUTPUT=`docker compose run --no-deps --rm test /scripts/sshrc_kill_test.sh`
 
 if [[ "$SSHRC_KILL_OUTPUT" == *"this output should never print"* ]]; then
   echo "${red}* Failure to quit after non-zero exit code in sshrc${reset}"