1 files changed, 2 insertions, 3 deletions
diff --git a/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c b/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
index 4d0c76f..b68c1db 100644
--- a/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
+++ b/MdeModulePkg/Universal/Disk/PartitionDxe/Partition.c
@@ -695,14 +695,13 @@ InitializePartition (
   //
   // Install driver model protocol(s).
   //
-  Status = EfiLibInstallAllDriverProtocols (
+  Status = EfiLibInstallDriverBindingComponentName2 (
              ImageHandle,
              SystemTable,
              &gPartitionDriverBinding,
              ImageHandle,
              &gPartitionComponentName,
-             NULL,
-             NULL
+             &gPartitionComponentName2
              );
   ASSERT_EFI_ERROR (Status);