diff --git a/src/kernel/boot/loader/heap.cpp b/src/kernel/boot/loader/heap.cpp
index 55e21bc851..fc11fdd08c 100644
--- a/src/kernel/boot/loader/heap.cpp
+++ b/src/kernel/boot/loader/heap.cpp
@@ -172,9 +172,9 @@ free_chunk::SetToAllocated(void *allocated)
 
 
 void
-heap_release(void)
+heap_release(stage2_args *args)
 {
-	platform_release_heap(sHeapBase);
+	platform_release_heap(args, sHeapBase);
 }