diff --git a/unit_tests/mocks/mock_memory_manager.cpp b/unit_tests/mocks/mock_memory_manager.cpp
index cdda0c45c5..2341f255f3 100644
--- a/unit_tests/mocks/mock_memory_manager.cpp
+++ b/unit_tests/mocks/mock_memory_manager.cpp
@@ -98,6 +98,13 @@ GraphicsAllocation *MockMemoryManager::allocate32BitGraphicsMemory(size_t size,
     return allocate32BitGraphicsMemoryImpl(allocationData);
 }
 
+GraphicsAllocation *MockMemoryManager::allocate32BitGraphicsMemoryImpl(const AllocationData &allocationData) {
+    if (failAllocate32Bit) {
+        return nullptr;
+    }
+    return OsAgnosticMemoryManager::allocate32BitGraphicsMemoryImpl(allocationData);
+}
+
 FailMemoryManager::FailMemoryManager(int32_t failedAllocationsCount, ExecutionEnvironment &executionEnvironment) : MockMemoryManager(executionEnvironment) {
     this->failedAllocationsCount = failedAllocationsCount;
 }
diff --git a/unit_tests/mocks/mock_memory_manager.h b/unit_tests/mocks/mock_memory_manager.h
index ff665cb6be..0e40564db3 100644
--- a/unit_tests/mocks/mock_memory_manager.h
+++ b/unit_tests/mocks/mock_memory_manager.h
@@ -37,6 +37,7 @@ class MockMemoryManager : public MemoryManagerCreate<OsAgnosticMemoryManager> {
     using MemoryManager::createStorageInfoFromProperties;
     using MemoryManager::getAllocationData;
     using MemoryManager::getBanksCount;
+    using MemoryManager::gfxPartition;
     using MemoryManager::localMemoryUsageBankSelector;
     using MemoryManager::multiContextResourceDestructor;
     using MemoryManager::registeredEngines;
@@ -97,6 +98,7 @@ class MockMemoryManager : public MemoryManagerCreate<OsAgnosticMemoryManager> {
     }
 
     GraphicsAllocation *allocate32BitGraphicsMemory(size_t size, const void *ptr, GraphicsAllocation::AllocationType allocationType);
+    GraphicsAllocation *allocate32BitGraphicsMemoryImpl(const AllocationData &allocationData) override;
 
     uint32_t freeGraphicsMemoryCalled = 0u;
     uint32_t unlockResourceCalled = 0u;
@@ -112,6 +114,7 @@ class MockMemoryManager : public MemoryManagerCreate<OsAgnosticMemoryManager> {
     bool allocateForImageCalled = false;
     bool failReserveAddress = false;
     bool failAllocateSystemMemory = false;
+    bool failAllocate32Bit = false;
     std::unique_ptr<ExecutionEnvironment> mockExecutionEnvironment;
 };