diff --git a/tests/dataset/NormalizationLayerDataset.h b/tests/dataset/NormalizationLayerDataset.h
index 7234f41..fc56694 100644
--- a/tests/dataset/NormalizationLayerDataset.h
+++ b/tests/dataset/NormalizationLayerDataset.h
@@ -35,7 +35,7 @@
 
 #ifdef BOOST
 #include "boost_wrapper.h"
-#endif
+#endif /* BOOST */
 
 namespace arm_compute
 {
@@ -96,4 +96,4 @@
 
 } // namespace test
 } // namespace arm_compute
-#endif //__ARM_COMPUTE_TEST_DATASET_NORMALIZATION_LAYER_DATASET_H__
+#endif /* __ARM_COMPUTE_TEST_DATASET_NORMALIZATION_LAYER_DATASET_H__ */