diff --git a/sycl/unittests/xpti_trace/QueueIDCheck.cpp b/sycl/unittests/xpti_trace/QueueIDCheck.cpp
index 43d113c2eea3c..c57c7091c1fe9 100644
--- a/sycl/unittests/xpti_trace/QueueIDCheck.cpp
+++ b/sycl/unittests/xpti_trace/QueueIDCheck.cpp
@@ -141,8 +141,8 @@ TEST_F(QueueID, QueueCreationUSMOperations) {
   auto Queue0ImplPtr = sycl::detail::getSyclObjImpl(Q0);
   auto QueueIDSTr = std::to_string(Queue0ImplPtr->getQueueID());
 
-  unsigned char *AllocSrc = (unsigned char *)sycl::malloc_shared(1, Q0);
-  unsigned char *AllocDst = (unsigned char *)sycl::malloc_shared(1, Q0);
+  unsigned char *AllocSrc = (unsigned char *)sycl::malloc_device(1, Q0);
+  unsigned char *AllocDst = (unsigned char *)sycl::malloc_device(1, Q0);
   Q0.memset(AllocSrc, 42, 1).wait();
   checkTaskBeginEnd(QueueIDSTr);