diff --git a/tests/dataset/GenericDataset.h b/tests/dataset/GenericDataset.h
index 4ca489b..48754fc 100644
--- a/tests/dataset/GenericDataset.h
+++ b/tests/dataset/GenericDataset.h
@@ -31,7 +31,7 @@
 
 #ifdef BOOST
 #include "boost_wrapper.h"
-#endif
+#endif /* BOOST */
 
 namespace arm_compute
 {
@@ -58,9 +58,9 @@
     /** Number of samples in the data set. */
 #ifdef BOOST
     boost::unit_test::data::size_t size() const
-#else
+#else  /* BOOST */
     unsigned int size() const
-#endif
+#endif /* BOOST */
     {
         return _data.size();
     }
@@ -94,4 +94,4 @@
 };
 } // namespace test
 } // namespace arm_compute
-#endif //__ARM_COMPUTE_TEST_DATASET_GENERIC_DATASET_H__
+#endif /* __ARM_COMPUTE_TEST_DATASET_GENERIC_DATASET_H__ */