diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index a54cd73..9f7b5cb 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1433,9 +1433,9 @@
 #if defined(static_assert)
 static_assert( ( MBEDTLS_PSA_KA_MASK_EXTERNAL_ONLY & MBEDTLS_PSA_KA_MASK_DUAL_USE ) == 0,
                "One or more key attribute flag is listed as both external-only and dual-use" );
-static_assert( ( MBEDTLS_PSA_KA_MASK_INTERNAL_ONLY & MBEDTLS_PSA_KA_MASK_DUAL_USE ) == 0,
+static_assert( ( PSA_KA_MASK_INTERNAL_ONLY & MBEDTLS_PSA_KA_MASK_DUAL_USE ) == 0,
                "One or more key attribute flag is listed as both external-only and dual-use" );
-static_assert( ( MBEDTLS_PSA_KA_MASK_INTERNAL_ONLY & MBEDTLS_PSA_KA_MASK_EXTERNAL_ONLY ) == 0,
+static_assert( ( PSA_KA_MASK_INTERNAL_ONLY & MBEDTLS_PSA_KA_MASK_EXTERNAL_ONLY ) == 0,
                "One or more key attribute flag is listed as both internal-only and external-only" );
 #endif
 
diff --git a/library/psa_crypto_core.h b/library/psa_crypto_core.h
index b67c0c5..edf3ab6 100644
--- a/library/psa_crypto_core.h
+++ b/library/psa_crypto_core.h
@@ -68,7 +68,7 @@
 
 /* A mask of key attribute flags used only internally.
  * Currently there aren't any. */
-#define MBEDTLS_PSA_KA_MASK_INTERNAL_ONLY (     \
+#define PSA_KA_MASK_INTERNAL_ONLY (     \
         0 )
 
 /** Test whether a key slot is occupied.