diff --git a/sycl/test-e2e/USM/dmem_varied.cpp b/sycl/test-e2e/USM/dmem_varied.cpp
index 440ffcfe4c09b..ef8bc916e5185 100644
--- a/sycl/test-e2e/USM/dmem_varied.cpp
+++ b/sycl/test-e2e/USM/dmem_varied.cpp
@@ -79,12 +79,12 @@ int main() {
   }
 
   q.submit([&](handler &h) {
-    h.single_task<class foo1>([=]() {
-      for (size_t i = 0; i < count; ++i) {
-        *ptrs[i] = 1;
-      }
-    });
-  });
+     h.single_task<class foo1>([=]() {
+       for (size_t i = 0; i < count; ++i) {
+         *ptrs[i] = 1;
+       }
+     });
+   }).wait();
 
   size_t *res =
       (size_t *)aligned_alloc_shared(alignof(size_t), sizeof(size_t), q);