diff --git a/tests/main.bats b/tests/main.bats
index 750f75b..411a763 100644
--- a/tests/main.bats
+++ b/tests/main.bats
@@ -51,8 +51,8 @@ load ../common
        curl -L -o ./clair-scanner https://github.com/arminc/clair-scanner/releases/download/v8/clair-scanner_linux_amd64
        chmod 755 clair-scanner
     fi
-    docker ps | grep db
-    if [ $? == "0" ]; then
+    run docker ps | grep db
+    if [ $status == "0" ]; then
       docker kill db
       docker rm db
       docker run -p 5432:5432 -d --name db arminc/clair-db:latest
@@ -60,8 +60,8 @@ load ../common
       docker run -p 5432:5432 -d --name db arminc/clair-db:latest
     fi
     sleep 15
-    docker ps | grep clair
-    if [ $? == "0" ]; then
+    run docker ps | grep clair
+    if [ $status == "0" ]; then
       docker kill clair
       docker rm clair
       docker run -p 6060:6060 --link db:postgres -d --name clair arminc/clair-local-scan:v2.0.5