diff --git a/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c b/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
index dd62171749..ed564e5cf3 100644
--- a/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
+++ b/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
@@ -1053,13 +1053,13 @@ PartitionInstallChildHandle (
   //  for logical partitions.
   //
   if (Private->BlockIo.Revision >= EFI_BLOCK_IO_PROTOCOL_REVISION2) {
-    Private->BlockIo.Media->LowestAlignedLba               = 0;
-    Private->BlockIo.Media->LogicalBlocksPerPhysicalBlock  = 0;
-    Private->BlockIo2.Media->LowestAlignedLba              = 0;
-    Private->BlockIo2.Media->LogicalBlocksPerPhysicalBlock = 0;
+    Private->Media.LowestAlignedLba               = 0;
+    Private->Media.LogicalBlocksPerPhysicalBlock  = 0;
+    Private->Media2.LowestAlignedLba              = 0;
+    Private->Media2.LogicalBlocksPerPhysicalBlock = 0;
     if (Private->BlockIo.Revision >= EFI_BLOCK_IO_PROTOCOL_REVISION3) {
-      Private->BlockIo.Media->OptimalTransferLengthGranularity  = 0;
-      Private->BlockIo2.Media->OptimalTransferLengthGranularity = 0;
+      Private->Media.OptimalTransferLengthGranularity  = 0;
+      Private->Media2.OptimalTransferLengthGranularity = 0;
     }
   }