diff --git a/library/arc4.c b/library/arc4.c
index dbf0c7e..464f800 100644
--- a/library/arc4.c
+++ b/library/arc4.c
@@ -178,6 +178,6 @@
     return( 0 );
 }
 
-#endif
+#endif /* POLARSSL_SELF_TEST */
 
-#endif
+#endif /* POLARSSL_ARC4_C */