diff --git a/library/md4.c b/library/md4.c
index aa5f539..5d3315f 100644
--- a/library/md4.c
+++ b/library/md4.c
@@ -407,7 +407,7 @@
  */
 static const char md4_test_str[7][81] =
 {
-    { "" }, 
+    { "" },
     { "a" },
     { "abc" },
     { "message digest" },
@@ -469,6 +469,6 @@
     return( 0 );
 }
 
-#endif
+#endif /* POLARSSL_SELF_TEST */
 
-#endif
+#endif /* POLARSSL_MD4_C */