fix various issues

Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>
diff --git a/library/pk.c b/library/pk.c
index 4c3c774..0a7a40b 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -545,14 +545,10 @@
                                  sig, sig_size, sig_len, f_rng, p_rng ) );
     }
 
-#if defined(MBEDTLS_X509_RSASSA_PSS_SUPPORT)
     return( mbedtls_pk_psa_sign_ext( PSA_ALG_RSA_PSS(
                                         mbedtls_psa_translate_md( md_alg ) ),
                                      ctx->pk_ctx, hash, hash_len,
                                      sig, sig_size, sig_len ) );
-#else /* MBEDTLS_X509_RSASSA_PSS_SUPPORT */
-    return( MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE );
-#endif /* !MBEDTLS_X509_RSASSA_PSS_SUPPORT */
 
 }
 #endif /* MBEDTLS_PSA_CRYPTO_C */
diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 4c66e5f..76b7a1b 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -251,7 +251,7 @@
         ret = mbedtls_pk_error_from_psa( status );
     return( ret );
 }
-#endif /* MBEDTLS_USE_PSA_CRYPTO */
+#endif /* MBEDTLS_PSA_CRYPTO_C */
 
 #if defined(MBEDTLS_USE_PSA_CRYPTO)
 static int rsa_sign_wrap( void *ctx, mbedtls_md_type_t md_alg,
diff --git a/library/pk_wrap.h b/library/pk_wrap.h
index d4a90ab..e310005 100644
--- a/library/pk_wrap.h
+++ b/library/pk_wrap.h
@@ -155,6 +155,6 @@
                              unsigned char *sig, size_t sig_size,
                              size_t *sig_len );
 
-#endif /* MBEDTLS_USE_PSA_CRYPTO */
+#endif /* MBEDTLS_PSA_CRYPTO_C */
 
 #endif /* MBEDTLS_PK_WRAP_H */
diff --git a/library/ssl_tls13_generic.c b/library/ssl_tls13_generic.c
index a8c3570..91c5448 100644
--- a/library/ssl_tls13_generic.c
+++ b/library/ssl_tls13_generic.c
@@ -1129,7 +1129,6 @@
         pk_type = MBEDTLS_PK_ECDSA;
         break;
 #endif /* MBEDTLS_ECDSA_C */
-#if defined(MBEDTLS_X509_RSASSA_PSS_SUPPORT)
     case MBEDTLS_TLS1_3_SIG_RSA_PSS_RSAE_SHA256:
         md_alg = MBEDTLS_MD_SHA256;
         pk_type = MBEDTLS_PK_RSASSA_PSS;
@@ -1142,7 +1141,6 @@
         md_alg = MBEDTLS_MD_SHA512;
         pk_type = MBEDTLS_PK_RSASSA_PSS;
         break;
-#endif /* MBEDTLS_X509_RSASSA_PSS_SUPPORT */
     default:
         break;
     }