1 files changed, 8 insertions, 8 deletions
diff --git a/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/InstallFdt.c b/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/InstallFdt.c
index a4220a1895..51ea7c00fe 100644
--- a/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/InstallFdt.c
+++ b/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/InstallFdt.c
@@ -174,14 +174,14 @@ JunoFdtStart (
   Status = BootMonFs->OpenVolume (BootMonFs, &Fs);
   if (EFI_ERROR (Status)) {
     PrintMessage ("Warning: Fail to open file system that should contain FDT file.\n");
-    goto UNLOAD_PROTOCOL;
+    goto CLOSE_PROTOCOL;
   }
 
   File = NULL;
   Status = Fs->Open (Fs, &File, mFdtFileName, EFI_FILE_MODE_READ, 0);
   if (EFI_ERROR (Status)) {
     PrintMessage ("Warning: Fail to load FDT file '%s'.\n", mFdtFileName);
-    goto UNLOAD_PROTOCOL;
+    goto CLOSE_PROTOCOL;
   }
 
   Size = 0;
@@ -189,7 +189,7 @@ JunoFdtStart (
   FileInfo = AllocatePool (Size);
   Status = File->GetInfo (File, &gEfiFileInfoGuid, &Size, FileInfo);
   if (EFI_ERROR (Status)) {
-    goto UNLOAD_PROTOCOL;
+    goto CLOSE_PROTOCOL;
   }
 
   // Get the file size
@@ -219,13 +219,13 @@ JunoFdtStart (
     }
   }
 
-UNLOAD_PROTOCOL:
+CLOSE_PROTOCOL:
   // We do not need the FileSystem protocol
   gBS->CloseProtocol (
-      ControllerHandle,
-      &gEfiSimpleFileSystemProtocolGuid,
-      DriverBinding->ImageHandle,
-      ControllerHandle);
+         ControllerHandle,
+         &gEfiSimpleFileSystemProtocolGuid,
+         gImageHandle,
+         ControllerHandle);
 
   return Status;
 }