diff --git a/include/psa/crypto.h b/include/psa/crypto.h
index 9f0b135..896235b 100644
--- a/include/psa/crypto.h
+++ b/include/psa/crypto.h
@@ -1174,6 +1174,7 @@
  * \retval #PSA_SUCCESS
  * \retval #PSA_ERROR_EMPTY_SLOT
  * \retval #PSA_ERROR_NOT_PERMITTED
+ * \retval #PSA_ERROR_NOT_SUPPORTED
  * \retval #PSA_ERROR_COMMUNICATION_FAILURE
  * \retval #PSA_ERROR_HARDWARE_FAILURE
  * \retval #PSA_ERROR_TAMPERING_DETECTED
diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 75c7ea8..959b9ec 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -787,16 +787,25 @@
         {
             mbedtls_pk_context pk;
             int ret;
-            mbedtls_pk_init( &pk );
             if( PSA_KEY_TYPE_IS_RSA( slot->type ) )
             {
+#if defined(MBEDTLS_RSA_C)
+                mbedtls_pk_init( &pk );
                 pk.pk_info = &mbedtls_rsa_info;
                 pk.pk_ctx = slot->data.rsa;
+#else
+                return( PSA_ERROR_NOT_SUPPORTED );
+#endif
             }
             else
             {
+#if defined(MBEDTLS_ECP_C)
+                mbedtls_pk_init( &pk );
                 pk.pk_info = &mbedtls_eckey_info;
                 pk.pk_ctx = slot->data.ecp;
+#else
+                return( PSA_ERROR_NOT_SUPPORTED );
+#endif
             }
             if( export_public_key || PSA_KEY_TYPE_IS_PUBLIC_KEY( slot->type ) )
                 ret = mbedtls_pk_write_pubkey_der( &pk, data, data_size );