diff --git a/library/pk.c b/library/pk.c
index 580fa0e..56f8bb1 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -1400,14 +1400,9 @@
         return MBEDTLS_ERR_PK_BAD_INPUT_DATA;
     }
 
-    if ((psa_get_key_usage_flags(&key_attr) & PSA_KEY_USAGE_EXPORT) != PSA_KEY_USAGE_EXPORT) {
-        ret = MBEDTLS_ERR_PK_BAD_INPUT_DATA;
-        goto exit;
-    }
-
     status = psa_export_key(key_id, exp_key, sizeof(exp_key), &exp_key_len);
     if (status != PSA_SUCCESS) {
-        ret = psa_generic_status_to_mbedtls(status);
+        ret = MBEDTLS_ERR_PK_BAD_INPUT_DATA;
         goto exit;
     }