X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fgrub2%2Fenv.c;h=1c8635fbc924427b880ab518ff3e3ba7664dc259;hp=90e9c69667abe47db82ee324c3f52052bba99648;hb=37bff93c8b0a71432613f41f2319dc073ca64619;hpb=ccacb83a030da564c59c8e0f081fe967fba5a3eb

diff --git a/discover/grub2/env.c b/discover/grub2/env.c
index 90e9c69..1c8635f 100644
--- a/discover/grub2/env.c
+++ b/discover/grub2/env.c
@@ -5,7 +5,7 @@
 #include <log/log.h>
 #include <types/types.h>
 #include <talloc/talloc.h>
-#include <array-size/array-size.h>
+#include <util/util.h>
 
 #include <discover/parser.h>
 #include <discover/file.h>
@@ -84,10 +84,10 @@ int builtin_load_env(struct grub2_script *script,
 
 	rc = parser_request_file(script->ctx, dev, envpath, &buf, &len);
 
-	if (!rc)
+	if (!rc) {
 		rc = parse_buf_to_env(script, buf, len);
-
-	talloc_free(buf);
+		talloc_free(buf);
+	}
 
 	return 0;
 }