diff --git a/shared/source/os_interface/linux/drm_neo.h b/shared/source/os_interface/linux/drm_neo.h
index 39d3253109..1428ed062f 100644
--- a/shared/source/os_interface/linux/drm_neo.h
+++ b/shared/source/os_interface/linux/drm_neo.h
@@ -108,7 +108,7 @@ class Drm {
     std::unique_ptr<MemoryInfo> memoryInfo;
 
     std::string getSysFsPciPath(int deviceID);
-    void *query(uint32_t queryId);
+    std::unique_ptr<uint8_t[]> query(uint32_t queryId);
 
 #pragma pack(1)
     struct PCIConfig {
diff --git a/shared/source/os_interface/linux/drm_query.cpp b/shared/source/os_interface/linux/drm_query.cpp
index 934a321942..404adefc54 100644
--- a/shared/source/os_interface/linux/drm_query.cpp
+++ b/shared/source/os_interface/linux/drm_query.cpp
@@ -11,7 +11,7 @@
 
 namespace NEO {
 
-void *Drm::query(uint32_t queryId) {
+std::unique_ptr<uint8_t[]> Drm::query(uint32_t queryId) {
     return nullptr;
 }