diff --git a/library/md2.c b/library/md2.c
index 1e1691d..af6b595 100644
--- a/library/md2.c
+++ b/library/md2.c
@@ -373,6 +373,6 @@
     return( 0 );
 }
 
-#endif
+#endif /* POLARSSL_SELF_TEST */
 
-#endif
+#endif /* POLARSSL_MD2_C */