Fix the build with MBEDTLS_RSA_ALT

Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
diff --git a/library/pk.c b/library/pk.c
index 60891a0..1911325 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -449,7 +449,7 @@
             psa_set_key_type(attributes, (want_private ?
                                           PSA_KEY_TYPE_RSA_KEY_PAIR :
                                           PSA_KEY_TYPE_RSA_PUBLIC_KEY));
-            psa_set_key_bits(attributes, mbedtls_mpi_bitlen(&rsa->N));
+            psa_set_key_bits(attributes, mbedtls_pk_get_bitlen(pk));
             psa_set_key_algorithm(attributes,
                                   psa_algorithm_for_rsa(rsa, want_crypt));
             break;