diff --git a/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/aie.mlir b/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/aie.mlir
index 8bfd010a808..c128724b0ca 100644
--- a/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/aie.mlir
+++ b/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/aie.mlir
@@ -22,12 +22,12 @@ module @test_chesss_01_precompiled_core_function {
   %lock13_3 = AIE.lock(%tile13, 3) { sym_name = "input_lock" }
   %lock13_5 = AIE.lock(%tile13, 5) { sym_name = "output_lock" }
 
-  func.func private @func(%A: memref<256xi32>, %B: memref<256xi32>) -> ()
+  func.func private @func(%A: memref<256xi32>, %B: memref<256xi32>, %lock: index) -> ()
 
   %core13 = AIE.core(%tile13) { 
     AIE.useLock(%lock13_3, "Acquire", 1) // acquire for read(e.g. input ping)
     AIE.useLock(%lock13_5, "Acquire", 0) // acquire for write
-    func.call @func(%buf13_0, %buf13_1) : (memref<256xi32>, memref<256xi32>) -> ()
+    func.call @func(%buf13_0, %buf13_1, %lock13_3) : (memref<256xi32>, memref<256xi32>, index) -> ()
     AIE.useLock(%lock13_3, "Release", 0) // release for write
     AIE.useLock(%lock13_5, "Release", 1) // release for read
     AIE.end
diff --git a/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.cc b/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.cc
index a94fbc7dcef..39a51c8932f 100644
--- a/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.cc
+++ b/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.cc
@@ -10,13 +10,15 @@
 
 #include "kernel.h"
 
-void func(int32_t *a, int32_t *b)
+void func(int32_t *a, int32_t *b, int64_t lock)
 {
+    acquire(lock, 1);
     int val=a[3];
     int val2=val+val;
     val2 += val;
     val2 += val;
     val2 += val;
     b[5] = val2;
+    release(lock, 0);
 }
 
diff --git a/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.h b/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.h
index 5b192c91eb3..78f63f86ad3 100644
--- a/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.h
+++ b/test/unit_tests/chess_compiler_tests/01_precompiled_core_function/kernel.h
@@ -11,5 +11,5 @@
 #include <stdio.h>
 
 extern "C" {
-void func(int32_t *a, int32_t *b);
+void func(int32_t *a, int32_t *b, int64_t lock);
 }