diff --git a/unit_tests/test_files/simple_kernels.cl b/unit_tests/test_files/simple_kernels.cl
index 7809fdb037..9944c4b2df 100644
--- a/unit_tests/test_files/simple_kernels.cl
+++ b/unit_tests/test_files/simple_kernels.cl
@@ -125,5 +125,6 @@ __kernel void simple_kernel_8(__global uint *dst, uint incrementationsCount) {
 }
 
 __kernel void simple_kernel_9(__global uint *dst) {
-    dst[get_sub_group_id()]++;
+    uint offset = get_max_sub_group_size() * get_sub_group_id();
+    dst[get_sub_group_local_id() + offset] = get_local_id(0);
 }