diff --git a/library/base64.c b/library/base64.c
index 103d814..d7f3642 100644
--- a/library/base64.c
+++ b/library/base64.c
@@ -267,6 +267,6 @@
     return( 0 );
 }
 
-#endif
+#endif /* POLARSSL_SELF_TEST */
 
-#endif
+#endif /* POLARSSL_BASE64_C */