1 files changed, 2 insertions, 3 deletions
diff --git a/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c b/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
index 0cb4922..8f50ec8 100644
--- a/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
+++ b/MdeModulePkg/Universal/Disk/DiskIoDxe/DiskIo.c
@@ -720,14 +720,13 @@ InitializeDiskIo (
   //
   // Install driver model protocol(s).
   //
-  Status = EfiLibInstallAllDriverProtocols (
+  Status = EfiLibInstallDriverBindingComponentName2 (
              ImageHandle,
              SystemTable,
              &gDiskIoDriverBinding,
              ImageHandle,
              &gDiskIoComponentName,
-             NULL,
-             NULL
+             &gDiskIoComponentName2
              );
   ASSERT_EFI_ERROR (Status);