diff --git a/tests/dataset/InterpolationPolicyDataset.h b/tests/dataset/InterpolationPolicyDataset.h
index e6062ea..0f3184c 100644
--- a/tests/dataset/InterpolationPolicyDataset.h
+++ b/tests/dataset/InterpolationPolicyDataset.h
@@ -28,7 +28,7 @@
 
 #ifdef BOOST
 #include "boost_wrapper.h"
-#endif
+#endif /* BOOST */
 
 namespace arm_compute
 {
@@ -54,9 +54,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 _policies.size();
     }
@@ -77,4 +77,4 @@
 };
 } // namespace test
 } // namespace arm_compute
-#endif
+#endif /* __ARM_COMPUTE_TEST_INTERPOLATION_POLICY_DATASET_H__ */