diff --git a/library/pkwrite.c b/library/pkwrite.c
index 741df08..022281d 100644
--- a/library/pkwrite.c
+++ b/library/pkwrite.c
@@ -25,7 +25,7 @@
 
 #include "polarssl/config.h"
 
-#if defined(POLARSSL_PK_C)
+#if defined(POLARSSL_PK_WRITE_C)
 
 #include "polarssl/pk.h"
 #include "polarssl/asn1write.h"
@@ -386,4 +386,4 @@
 }
 #endif /* POLARSSL_BASE64_C */
 
-#endif /* POLARSSL_PK_C */
+#endif /* POLARSSL_PK_WRITE_C */