X-Git-Url: https://code.delx.au/refind/blobdiff_plain/0d5ee58bd6562186ea980fcca99484d8272369bf..4c9f41e161bd197922912efbcf4cc676077d5c00:/libeg/image.c

diff --git a/libeg/image.c b/libeg/image.c
index 271141f..28d2472 100644
--- a/libeg/image.c
+++ b/libeg/image.c
@@ -37,10 +37,16 @@
 #include "libegint.h"
 #include "../refind/global.h"
 #include "../refind/lib.h"
-#include "refit_call_wrapper.h"
+#include "../refind/screen.h"
+#include "../include/refit_call_wrapper.h"
 
 #define MAX_FILE_SIZE (1024*1024*1024)
 
+#ifndef __MAKEWITH_GNUEFI
+#define LibLocateHandle gBS->LocateHandleBuffer
+#define LibOpenRoot EfiLibOpenRoot
+#endif
+
 //
 // Basic image handling
 //