diff --git a/opencl/test/unit_test/linux/mock_os_layer.cpp b/opencl/test/unit_test/linux/mock_os_layer.cpp
index 787b4461a4..4a85204799 100644
--- a/opencl/test/unit_test/linux/mock_os_layer.cpp
+++ b/opencl/test/unit_test/linux/mock_os_layer.cpp
@@ -215,7 +215,6 @@ int drmContextDestroy(drm_i915_gem_context_destroy *destroy) {
 
 int drmVirtualMemoryCreate(drm_i915_gem_vm_control *control) {
     assert(control);
-
     control->vm_id = ++vmId;
     return failOnVirtualMemoryCreate;
 }
diff --git a/opencl/test/unit_test/os_interface/linux/sys_calls_linux_ult.cpp b/opencl/test/unit_test/os_interface/linux/sys_calls_linux_ult.cpp
index 775db35e19..bac45386b4 100644
--- a/opencl/test/unit_test/os_interface/linux/sys_calls_linux_ult.cpp
+++ b/opencl/test/unit_test/os_interface/linux/sys_calls_linux_ult.cpp
@@ -30,6 +30,9 @@ constexpr int fakeFileDescriptor = 123;
 uint32_t vmId = 0;
 bool makeFakeDevicePath = false;
 bool allowFakeDevicePath = false;
+uint32_t ioctlVmCreateCalled = 0u;
+int ioctlVmCreateReturned = 0u;
+uint64_t ioctlVmCreateExtensionArg = 0ull;
 
 int close(int fileDescriptor) {
     closeFuncCalled++;
@@ -61,9 +64,11 @@ int ioctl(int fileDescriptor, unsigned long int request, void *arg) {
         }
     }
     if (request == DRM_IOCTL_I915_GEM_VM_CREATE) {
+        ioctlVmCreateCalled++;
         auto control = static_cast<drm_i915_gem_vm_control *>(arg);
+        ioctlVmCreateExtensionArg = control->extensions;
         control->vm_id = ++vmId;
-        return 0;
+        return ioctlVmCreateReturned;
     }
     if (request == DRM_IOCTL_I915_GEM_VM_DESTROY) {
         auto control = static_cast<drm_i915_gem_vm_control *>(arg);