Update to support to produce Component Name and & Component Name 2 protocol based...

[mirror_edk2.git] / MdeModulePkg / Universal / Disk / DiskIoDxe / DiskIo.c

diff --git a/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c b/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
index 0cb49224ca7ad80c13ad43cc0340482a05378976..8f50ec859df5227befd773b5af843e7d680f4c4c 100644 (file)
--- a/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
+++ b/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
@@ -720,14 +720,13 @@ InitializeDiskIo (
   //\r
   // Install driver model protocol(s).\r
   //\r
-  Status = EfiLibInstallAllDriverProtocols (\r
+  Status = EfiLibInstallDriverBindingComponentName2 (\r
              ImageHandle,\r
              SystemTable,\r
              &gDiskIoDriverBinding,\r
              ImageHandle,\r
              &gDiskIoComponentName,\r
-             NULL,\r
-             NULL\r
+             &gDiskIoComponentName2\r
              );\r
   ASSERT_EFI_ERROR (Status);\r
 \r