diff --git a/clang/test/dpct/query_api_mapping/Runtime/test_memory_management.cu b/clang/test/dpct/query_api_mapping/Runtime/test_memory_management.cu
index a61d62a10547..0e509bddf17a 100644
--- a/clang/test/dpct/query_api_mapping/Runtime/test_memory_management.cu
+++ b/clang/test/dpct/query_api_mapping/Runtime/test_memory_management.cu
@@ -46,7 +46,7 @@
 // CUDAGETSYMBOLADDRESS: CUDA API:
 // CUDAGETSYMBOLADDRESS-NEXT:   cudaGetSymbolAddress(pDev /*void ***/, symbol /*const void **/);
 // CUDAGETSYMBOLADDRESS-NEXT: Is migrated to:
-// CUDAGETSYMBOLADDRESS-NEXT:   *(pDev) = const_cast<void *>(symbol);
+// CUDAGETSYMBOLADDRESS-NEXT:   *(pDev) = symbol.get_ptr();
 
 // RUN: dpct --cuda-include-path="%cuda-path/include" --query-api-mapping=cudaGetSymbolSize | FileCheck %s -check-prefix=CUDAGETSYMBOLSIZE
 // CUDAGETSYMBOLSIZE: CUDA API: