summaryrefslogtreecommitdiff
path: root/replacements.c
diff options
context:
space:
mode:
Diffstat (limited to 'replacements.c')
-rw-r--r--replacements.c7
1 files changed, 2 insertions, 5 deletions
diff --git a/replacements.c b/replacements.c
index c64c8d13..278a8e78 100644
--- a/replacements.c
+++ b/replacements.c
@@ -18,11 +18,6 @@
* National Security Policy and Scientific Developments, November 20,
* 1969.
*/
-
-#include <efi.h>
-#include <efiapi.h>
-#include <efilib.h>
-
#include "shim.h"
static EFI_SYSTEM_TABLE *systab;
@@ -133,6 +128,8 @@ do_exit(EFI_HANDLE ImageHandle, EFI_STATUS ExitStatus,
shim_fini();
+ restore_loaded_image();
+
efi_status = gBS->Exit(ImageHandle, ExitStatus,
ExitDataSize, ExitData);
if (EFI_ERROR(efi_status)) {