diff --git a/programs/test/selftest.c b/programs/test/selftest.c
index 0e906ab4a3..04d35cd8ed 100644
--- a/programs/test/selftest.c
+++ b/programs/test/selftest.c
@@ -289,10 +289,7 @@ const selftest_t selftests[] =
 #if defined(MBEDTLS_SHA512_C)
     { "sha512", mbedtls_sha512_self_test },
 #endif
-#if defined(PSA_WANT_ALG_SHA3_224) || \
-    defined(PSA_WANT_ALG_SHA3_256) || \
-    defined(PSA_WANT_ALG_SHA3_384) || \
-    defined(PSA_WANT_ALG_SHA3_512)
+#if defined(MBEDTLS_SHA3_C)
     { "sha3", mbedtls_sha3_self_test },
 #endif
 #if defined(MBEDTLS_AES_C)