diff --git a/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c b/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
index 929f38343b..2ecf5b6605 100644
--- a/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
+++ b/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
@@ -614,6 +614,7 @@ Var_UpdateDriverOption (
 
   if (OptionalDataExist) {
     NewLoadContext->OptionalData = AllocateZeroPool (LoadOption.OptionalDataSize);
+    ASSERT (NewLoadContext->OptionalData != NULL);
     CopyMem (
       NewLoadContext->OptionalData,
       LoadOption.OptionalData,
@@ -725,6 +726,7 @@ Var_UpdateBootOption (
 
   if (OptionalDataExist) {
     NewLoadContext->OptionalData = AllocateZeroPool (LoadOption.OptionalDataSize);
+    ASSERT (NewLoadContext->OptionalData != NULL);
     CopyMem (
       NewLoadContext->OptionalData,
       LoadOption.OptionalData,