diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 2b6c8d42cb..959b94536e 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -261,7 +261,7 @@ psa_status_t mbedtls_to_psa_error(int ret)
             return PSA_ERROR_INSUFFICIENT_ENTROPY;
 #endif
 
-#if defined(MBEDTLS_MD_C)
+#if defined(MBEDTLS_MD_LIGHT)
         case MBEDTLS_ERR_MD_FEATURE_UNAVAILABLE:
             return PSA_ERROR_NOT_SUPPORTED;
         case MBEDTLS_ERR_MD_BAD_INPUT_DATA: