diff --git a/sycl/unittests/helpers/PiMockPlugin.hpp b/sycl/unittests/helpers/PiMockPlugin.hpp
index 75639bf56bbc2..f12917b1e70b9 100644
--- a/sycl/unittests/helpers/PiMockPlugin.hpp
+++ b/sycl/unittests/helpers/PiMockPlugin.hpp
@@ -1279,11 +1279,13 @@ mock_piextCommandBufferCreate(pi_context context, pi_device device,
 
 inline pi_result
 mock_piextCommandBufferRetain(pi_ext_command_buffer command_buffer) {
+  retainDummyHandle(command_buffer);
   return PI_SUCCESS;
 }
 
 inline pi_result
 mock_piextCommandBufferRelease(pi_ext_command_buffer command_buffer) {
+  releaseDummyHandle(command_buffer);
   return PI_SUCCESS;
 }