diff --git a/src/consts.rs b/src/consts.rs
index 743bee5642a4efdd038ec947dc462f1c3d4c6080..149c9b45764961ee0bbd1b797e43522545a8648a 100644
--- a/src/consts.rs
+++ b/src/consts.rs
@@ -15,7 +15,10 @@
     /// Offset to kernel heap
     pub const KERNEL_HEAP_OFFSET: usize = KERNEL_OFFSET + PML4_SIZE/2;
     /// Size of kernel heap
-    pub const KERNEL_HEAP_SIZE: usize = 256 * 1024 * 1024; // 256 MB
+    #[cfg(not(feature = "live"))]
+    pub const KERNEL_HEAP_SIZE: usize = 128 * 1024 * 1024; // 128 MB
+    #[cfg(feature = "live")]
+    pub const KERNEL_HEAP_SIZE: usize = 640 * 1024 * 1024; // 640 MB - 128 default + 512 for the live disk
 
     /// Offset to kernel percpu variables
     //TODO: Use 64-bit fs offset to enable this pub const KERNEL_PERCPU_OFFSET: usize = KERNEL_HEAP_OFFSET - PML4_SIZE;