diff --git a/SYCL/Basic/parallel_for_range.cpp b/SYCL/Basic/parallel_for_range.cpp
index c47ba48a37..811a4a96ab 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
@@ -87,8 +83,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) {