diff --git a/SYCL/Basic/parallel_for_range.cpp b/SYCL/Basic/parallel_for_range.cpp
index b2c70cd384..16d04d725c 100644
--- a/SYCL/Basic/parallel_for_range.cpp
+++ b/SYCL/Basic/parallel_for_range.cpp
@@ -13,10 +13,6 @@
 
 using namespace cl::sycl;
 
-[[cl::reqd_work_group_size(4, 4, 4)]] void reqd_wg_size_helper() {
-  // do nothing
-}
-
 int main() {
   auto AsyncHandler = [](exception_list ES) {
     for (auto &E : ES) {
@@ -48,8 +44,8 @@ int main() {
       try {
         Q.submit([&](handler &CGH) {
           CGH.parallel_for<class ReqdWGSizeNegativeA>(
-              nd_range<3>(range<3>(16, 16, 16), range<3>(8, 8, 8)),
-              [=](nd_item<3>) { reqd_wg_size_helper(); });
+	      nd_range<3>(range<3>(16, 16, 16), range<3>(8, 8, 8)), [=
+          ](nd_item<3>) [[sycl::reqd_work_group_size(4, 4, 4)]]{});
         });
         Q.wait_and_throw();
         std::cerr
@@ -83,8 +79,8 @@ int main() {
       // -> fail
       try {
         Q.submit([&](handler &CGH) {
-          CGH.parallel_for<class ReqdWGSizeNegativeB>(
-              range<3>(16, 16, 16), [=](item<3>) { reqd_wg_size_helper(); });
+	  CGH.parallel_for<class ReqdWGSizeNegativeB>(range<3>(16, 16, 16), [=
+          ](item<3>) [[sycl::reqd_work_group_size(4, 4, 4)]]{});
         });
         Q.wait_and_throw();
         std::cerr
@@ -123,8 +119,8 @@ int main() {
     try {
       Q.submit([&](handler &CGH) {
         CGH.parallel_for<class ReqdWGSizePositiveA>(
-            nd_range<3>(range<3>(8, 8, 8), range<3>(4, 4, 4)),
-            [=](nd_item<3>) { reqd_wg_size_helper(); });
+            nd_range<3>(range<3>(8, 8, 8), range<3>(4, 4, 4)), [=
+        ](nd_item<3>) [[sycl::reqd_work_group_size(4, 4, 4)]]{});
       });
       Q.wait_and_throw();
     } catch (nd_range_error &E) {