diff --git a/library/pkparse.c b/library/pkparse.c
index f03ace2..78eb4bc 100644
--- a/library/pkparse.c
+++ b/library/pkparse.c
@@ -34,7 +34,7 @@
 #include "mbedtls/rsa.h"
 #endif
 #include "mbedtls/ecp.h"
-#if defined(MBEDTLS_RSA_C) || defined(MBEDTLS_ECP_C)
+#if defined(MBEDTLS_RSA_C) || defined(MBEDTLS_PK_HAVE_ECC_KEYS)
 #include "pkwrite.h"
 #endif
 #if defined(MBEDTLS_PK_HAVE_ECC_KEYS)