diff --git a/openmp/libomptarget/test/mapping/target_has_device_addr.c b/openmp/libomptarget/test/mapping/target_has_device_addr.c
index 8d206167b5d54..dcb8f26b62487 100644
--- a/openmp/libomptarget/test/mapping/target_has_device_addr.c
+++ b/openmp/libomptarget/test/mapping/target_has_device_addr.c
@@ -11,8 +11,9 @@
 #define LENGTH 128
 
 void foo() {
+  const int device_id = omp_get_default_device();
   float *A;
-#pragma omp allocate(A) allocator(llvm_omp_target_shared_mem_alloc)
+  A = (float *)omp_target_alloc(sizeof(float), device_id);
 
   float *A_dev = NULL;
 #pragma omp target has_device_addr(A [FROM:LENGTH]) map(A_dev)