diff --git a/third_party/aub_stream/headers/aub_manager.h b/third_party/aub_stream/headers/aub_manager.h
index 4c7c7d04a1..eb75ffcba1 100644
--- a/third_party/aub_stream/headers/aub_manager.h
+++ b/third_party/aub_stream/headers/aub_manager.h
@@ -29,7 +29,7 @@ class AubManager {
 
     virtual void addComment(const char *message) = 0;
     virtual void writeMemory(uint64_t gfxAddress, const void *memory, size_t size, uint32_t memoryBanks,int hint, size_t pageSize) = 0;
-    virtual void writePageTableEntries(uint64_t gfxAddress, const void *memory, size_t size, uint32_t memoryBanks, int hint,
+    virtual void writePageTableEntries(uint64_t gfxAddress, size_t size, uint32_t memoryBanks, int hint,
                                        std::vector<PageInfo> &lastLevelPages, size_t pageSize) = 0;
 
     virtual void writePhysicalMemoryPages(const void *memory, std::vector<PageInfo> &pages, size_t size, int hint) = 0;
diff --git a/unit_tests/mocks/mock_aub_manager.h b/unit_tests/mocks/mock_aub_manager.h
index 98756ee6aa..77ef49464e 100644
--- a/unit_tests/mocks/mock_aub_manager.h
+++ b/unit_tests/mocks/mock_aub_manager.h
@@ -101,7 +101,7 @@ class MockAubManager : public aub_stream::AubManager {
         writeMemoryPageSizePassed = pageSize;
     }
 
-    void writePageTableEntries(uint64_t gfxAddress, const void *memory, size_t size, uint32_t memoryBanks, int hint,
+    void writePageTableEntries(uint64_t gfxAddress, size_t size, uint32_t memoryBanks, int hint,
                                std::vector<PageInfo> &lastLevelPages, size_t pageSize) override {
         writePageTableEntriesCalled = true;
     }