diff --git a/unit_tests/context/get_supported_image_formats_tests.cpp b/unit_tests/context/get_supported_image_formats_tests.cpp
index 496f4e81a7..97ac2805fc 100644
--- a/unit_tests/context/get_supported_image_formats_tests.cpp
+++ b/unit_tests/context/get_supported_image_formats_tests.cpp
@@ -181,6 +181,7 @@ TEST_P(PackedYuvExtensionSupportedImageFormatsTest, retrieveImageFormatsPackedYU
     bool isReadOnly = false;
     std::tie(imageFormatsFlags, imageFormats) = GetParam();
 
+    device->deviceInfo.nv12Extension = false;
     device->deviceInfo.packedYuvExtension = true;
 
     retVal = context->getSupportedImageFormats(
@@ -266,6 +267,7 @@ TEST_P(NV12ExtensionSupportedImageFormatsTest, givenNV12ExtensionWhenQueriedForI
     std::tie(imageFormatsFlags, imageFormats) = GetParam();
 
     device->deviceInfo.nv12Extension = true;
+    device->deviceInfo.packedYuvExtension = false;
 
     retVal = context->getSupportedImageFormats(
         device.get(),
diff --git a/unit_tests/mocks/mock_device.cpp b/unit_tests/mocks/mock_device.cpp
index 9da3c04682..e5506a25da 100644
--- a/unit_tests/mocks/mock_device.cpp
+++ b/unit_tests/mocks/mock_device.cpp
@@ -25,6 +25,7 @@ MockDevice::MockDevice()
     this->executionEnvironment->memoryManager = std::move(this->mockMemoryManager);
     this->engines.resize(defaultEngineIndex + 1);
     this->engines[defaultEngineIndex] = {commandStreamReceiver, nullptr};
+    initializeCaps();
 }
 
 MockDevice::MockDevice(ExecutionEnvironment *executionEnvironment, uint32_t deviceIndex)
@@ -36,6 +37,7 @@ MockDevice::MockDevice(ExecutionEnvironment *executionEnvironment, uint32_t devi
     this->osTime = MockOSTime::create();
     executionEnvironment->setHwInfo(&hwInfo);
     executionEnvironment->initializeMemoryManager();
+    initializeCaps();
 }
 
 void MockDevice::setOSTime(OSTime *osTime) {