diff --git a/media_driver/agnostic/common/codec/hal/codechal_decode_histogram_vebox.cpp b/media_driver/agnostic/common/codec/hal/codechal_decode_histogram_vebox.cpp
index b55b438f6bc..4552e38eeb9 100644
--- a/media_driver/agnostic/common/codec/hal/codechal_decode_histogram_vebox.cpp
+++ b/media_driver/agnostic/common/codec/hal/codechal_decode_histogram_vebox.cpp
@@ -37,7 +37,11 @@ CodechalDecodeHistogramVebox::CodechalDecodeHistogramVebox(
     MOS_ZeroMemory(&m_resStatisticsOutput, sizeof(m_resStatisticsOutput));
     MOS_ZeroMemory(&m_outputSurface, sizeof(m_outputSurface));
     // allocate heap
-    m_veboxInterface->CreateHeap();
+    MOS_STATUS status = m_veboxInterface->CreateHeap();
+    if (status != MOS_STATUS_SUCCESS)
+    {
+        CODECHAL_DECODE_ASSERTMESSAGE("Failed to CreateHeap.");
+    }
 
     // create Vebox context
     MOS_GPUCTX_CREATOPTIONS  createOpts;
diff --git a/media_driver/agnostic/common/shared/mediacopy/media_vebox_copy.cpp b/media_driver/agnostic/common/shared/mediacopy/media_vebox_copy.cpp
index 22ea3beecc0..508083453dc 100644
--- a/media_driver/agnostic/common/shared/mediacopy/media_vebox_copy.cpp
+++ b/media_driver/agnostic/common/shared/mediacopy/media_vebox_copy.cpp
@@ -75,7 +75,7 @@ MOS_STATUS VeboxCopyState::Initialize()
     {
         if (m_veboxInterface->m_veboxHeap == nullptr)
         {
-            m_veboxInterface->CreateHeap();
+            VEBOX_COPY_CHK_STATUS_RETURN(m_veboxInterface->CreateHeap());
         }
     }
     return MOS_STATUS_SUCCESS;
diff --git a/media_driver/agnostic/common/vp/hal/vphal_render_vebox_memdecomp.cpp b/media_driver/agnostic/common/vp/hal/vphal_render_vebox_memdecomp.cpp
index fde9c80a02a..83f6dac97ec 100644
--- a/media_driver/agnostic/common/vp/hal/vphal_render_vebox_memdecomp.cpp
+++ b/media_driver/agnostic/common/vp/hal/vphal_render_vebox_memdecomp.cpp
@@ -623,7 +623,7 @@ MOS_STATUS MediaVeboxDecompState::Initialize(
 
         if (m_veboxInterface->m_veboxHeap == nullptr)
         {
-            m_veboxInterface->CreateHeap();
+            VPHAL_MEMORY_DECOMP_CHK_STATUS_RETURN(m_veboxInterface->CreateHeap());
         }
 
 #if (_DEBUG || _RELEASE_INTERNAL)