diff --git a/tests/scripts/all.sh b/tests/scripts/all.sh
index 36a20b3..b9ff1fb 100755
--- a/tests/scripts/all.sh
+++ b/tests/scripts/all.sh
@@ -132,7 +132,7 @@
 }
 
 pre_check_environment () {
-    if [ \( -d library -o -d core \) -a -d include -a -d tests ]; then :; else
+    if in_mbedtls_repo || in_psa_crypto_repo; then :; else
         echo "Must be run from Mbed TLS root" >&2
         exit 1
     fi