diff --git a/tests/ha/wait_gateways.sh b/tests/ha/wait_gateways.sh
index d956b334..6edc2f6c 100755
--- a/tests/ha/wait_gateways.sh
+++ b/tests/ha/wait_gateways.sh
@@ -1,3 +1,5 @@
+#!/bin/sh
+set -ex
 SCALE=2
 echo CLI_TLS_ARGS $CLI_TLS_ARGS
 # Check if argument is provided
@@ -16,7 +18,7 @@ for i in $(seq $SCALE); do
     sleep 1  # Adjust the sleep duration as needed
     GW_NAME=$(docker ps --format '{{.ID}}\t{{.Names}}' | awk '$2 ~ /nvmeof/ && $2 ~ /'$i'/ {print $1}')
     container_status=$(docker inspect -f '{{.State.Status}}' "$GW_NAME")
-    if [ "$container_status" == "running" ]; then
+    if [ "$container_status" = "running" ]; then
       echo "Container $i $GW_NAME is now running."
     else
       echo "Container $i $GW_NAME is still not running. Waiting..."