diff --git a/fuchsia/test/va-test.cc b/fuchsia/test/va-test.cc
index 2b91e27..e8f7893 100644
--- a/fuchsia/test/va-test.cc
+++ b/fuchsia/test/va-test.cc
@@ -46,7 +46,7 @@
 
       {
         uint64_t vendor_id;
-        magma_status_t magma_status = magma_query2(device_, MAGMA_QUERY_VENDOR_ID, &vendor_id);
+        magma_status_t magma_status = magma_query(device_, MAGMA_QUERY_VENDOR_ID, nullptr, &vendor_id);
         if (magma_status == MAGMA_STATUS_OK && vendor_id == vendor_id_) {
           break;
         }
diff --git a/media_driver/linux/common/os/magma/mos_bufmgr_magma.cpp b/media_driver/linux/common/os/magma/mos_bufmgr_magma.cpp
index b7e46c0..b8cac48 100644
--- a/media_driver/linux/common/os/magma/mos_bufmgr_magma.cpp
+++ b/media_driver/linux/common/os/magma/mos_bufmgr_magma.cpp
@@ -534,9 +534,9 @@
     uint32_t device_id;
     {
         uint64_t val;
-        magma_status_t status = magma_query2(unowned_device, MAGMA_QUERY_DEVICE_ID, &val);
+        magma_status_t status = magma_query(unowned_device, MAGMA_QUERY_DEVICE_ID, nullptr, &val);
         if (status != MAGMA_STATUS_OK) {
-            LOG_VERBOSE("magma_query2 failed: %d", status);
+            LOG_VERBOSE("magma_query failed: %d", status);
             return nullptr;
         }
         device_id = static_cast<uint32_t>(val);
diff --git a/media_driver/linux/common/os/magma_fd.cpp b/media_driver/linux/common/os/magma_fd.cpp
index 334bd03..a0866f2 100644
--- a/media_driver/linux/common/os/magma_fd.cpp
+++ b/media_driver/linux/common/os/magma_fd.cpp
@@ -94,7 +94,7 @@
       return true;
 
     case I915_PARAM_CHIPSET_ID:
-      status = magma_query2(device, MAGMA_QUERY_DEVICE_ID, &val64);
+      status = magma_query(device, MAGMA_QUERY_DEVICE_ID, nullptr, &val64);
       if (status != MAGMA_STATUS_OK)
         return false;
       *value_out = static_cast<uint32_t>(val64);