1 files changed, 4 insertions, 2 deletions
diff --git a/OptionRomPkg/UndiRuntimeDxe/Init.c b/OptionRomPkg/UndiRuntimeDxe/Init.c
index 49ec0d5ded..3d0a3de5cb 100644
--- a/OptionRomPkg/UndiRuntimeDxe/Init.c
+++ b/OptionRomPkg/UndiRuntimeDxe/Init.c
@@ -1015,11 +1015,13 @@ InitializeUndi(
   EFI_EVENT     Event;
   EFI_STATUS    Status;
 
-  Status = EfiLibInstallDriverBinding (
+  Status = EfiLibInstallDriverBindingComponentName2 (
              ImageHandle,
              SystemTable,
              &gUndiDriverBinding,
-             ImageHandle
+             ImageHandle,
+             &gUndiComponentName,
+             &gUndiComponentName2
              );
   ASSERT_EFI_ERROR (Status);