diff --git a/library/oid.c b/library/oid.c
index e532bf3..e7c1224 100644
--- a/library/oid.c
+++ b/library/oid.c
@@ -26,7 +26,7 @@
 #include "mbedtls/oid.h"
 #include "mbedtls/rsa.h"
 #include "mbedtls/error.h"
-#include "pk_wrap.h"
+#include "mbedtls/pk.h"
 
 #include "mbedtls/legacy_or_psa.h"
 
diff --git a/library/pk_wrap.h b/library/pk_wrap.h
index ac5c72f..7df9608 100644
--- a/library/pk_wrap.h
+++ b/library/pk_wrap.h
@@ -31,10 +31,6 @@
 #include "psa/crypto.h"
 #endif /* MBEDTLS_PSA_CRYPTO_C */
 
-#if defined(MBEDTLS_PK_CAN_ECDSA_VERIFY) || defined(MBEDTLS_PK_CAN_ECDSA_SIGN)
-#define MBEDTLS_PK_CAN_ECDSA_SOME
-#endif
-
 struct mbedtls_pk_info_t {
     /** Public key type */
     mbedtls_pk_type_t type;