diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 9ef3f1e067..456d4e38f3 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -351,7 +351,7 @@ psa_status_t mbedtls_to_psa_error(int ret)
             return PSA_ERROR_INSUFFICIENT_ENTROPY;
 #endif
 
-#if defined(MBEDTLS_ECP_C)
+#if defined(MBEDTLS_ECP_LIGHT)
         case MBEDTLS_ERR_ECP_BAD_INPUT_DATA:
         case MBEDTLS_ERR_ECP_INVALID_KEY:
             return PSA_ERROR_INVALID_ARGUMENT;