diff --git a/tests/dataset/NonLinearFilterFunctionDataset.h b/tests/dataset/NonLinearFilterFunctionDataset.h
index 1778873..a2db323 100644
--- a/tests/dataset/NonLinearFilterFunctionDataset.h
+++ b/tests/dataset/NonLinearFilterFunctionDataset.h
@@ -28,7 +28,7 @@
 
 #ifdef BOOST
 #include "boost_wrapper.h"
-#endif
+#endif /* BOOST */
 
 #include <array>
 
@@ -56,9 +56,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 _functions.size();
     }
@@ -79,4 +79,4 @@
 };
 } // namespace test
 } // namespace arm_compute
-#endif
+#endif /* __ARM_COMPUTE_TEST_NON_LINEAR_FILTER_FUNCTION_DATASET_H__ */