1 files changed, 2 insertions, 5 deletions
diff --git a/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c b/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c
index 5903b0f641..ee10655244 100644
--- a/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c
+++ b/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c
@@ -1989,13 +1989,10 @@ S3BootScriptCompare (
   if (RelativePosition == NULL) {
     return EFI_INVALID_PARAMETER;
   }
-  TableLength = ((EFI_BOOT_SCRIPT_TABLE_HEADER*)Script)->TableLength;
   //
-  // If in boot time, TableLength does not include the termination node. so add it up 
+  // mS3BootScriptTablePtr->TableLength does not include the termination node, so add it up
   //
-  if (!mS3BootScriptTablePtr->AtRuntime) {
-    TableLength += sizeof(EFI_BOOT_SCRIPT_TERMINATE);
-  }
+  TableLength = mS3BootScriptTablePtr->TableLength + sizeof (EFI_BOOT_SCRIPT_TERMINATE);
   if (Position1 < Script || Position1 > Script+TableLength) {
     return EFI_INVALID_PARAMETER;
   }