diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index dce8e09..3722987 100755
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1349,7 +1349,8 @@
         return(PSA_ERROR_NOT_PERMITTED);
 
  #if defined(MBEDTLS_RSA_C)
-    if( slot->type == PSA_KEY_TYPE_RSA_KEYPAIR )
+    if( ( slot->type == PSA_KEY_TYPE_RSA_KEYPAIR ) ||
+        ( slot->type == PSA_KEY_TYPE_RSA_PUBLIC_KEY ) )
     {
         mbedtls_rsa_context *rsa = slot->data.rsa;
         int ret;
@@ -1438,7 +1439,8 @@
 
 
 #if defined(MBEDTLS_RSA_C)
-    if( slot->type == PSA_KEY_TYPE_RSA_KEYPAIR )
+    if( ( slot->type == PSA_KEY_TYPE_RSA_KEYPAIR ) ||
+        ( slot->type == PSA_KEY_TYPE_RSA_PUBLIC_KEY) )
     {
         mbedtls_rsa_context *rsa = slot->data.rsa;
         int ret;