diff --git a/headers/private/kernel/boot/heap.h b/headers/private/kernel/boot/heap.h
index b17f7a5f4d..fad877baa6 100644
--- a/headers/private/kernel/boot/heap.h
+++ b/headers/private/kernel/boot/heap.h
@@ -12,7 +12,7 @@
 extern "C" {
 #endif
 
-extern void heap_release(void);
+extern void heap_release(struct stage2_args *args);
 extern status_t heap_init(struct stage2_args *args);
 
 #ifdef __cplusplus
diff --git a/headers/private/kernel/boot/platform.h b/headers/private/kernel/boot/platform.h
index d9b8c73d49..7b39f5c46e 100644
--- a/headers/private/kernel/boot/platform.h
+++ b/headers/private/kernel/boot/platform.h
@@ -21,7 +21,7 @@ extern void panic(const char *format, ...);
 extern void dprintf(const char *format, ...);
 
 /* heap functions */
-extern void platform_release_heap(void *base);
+extern void platform_release_heap(struct stage2_args *args, void *base);
 extern status_t platform_init_heap(struct stage2_args *args, void **_base, void **_top);
 
 /* misc functions */
diff --git a/headers/private/kernel/boot/stage2_args.h b/headers/private/kernel/boot/stage2_args.h
index fb5488cee9..7d6c3597e1 100644
--- a/headers/private/kernel/boot/stage2_args.h
+++ b/headers/private/kernel/boot/stage2_args.h
@@ -5,10 +5,13 @@
 #ifndef KERNEL_BOOT_STAGE2_ARGS_H
 #define KERNEL_BOOT_STAGE2_ARGS_H
 
+
 #include <SupportDefs.h>
 #include <platform_stage2_args.h>
 
+
 typedef struct stage2_args {
+	size_t heap_size;
 	struct platform_stage2_args	platform;
 } stage2_args ;