diff --git a/rialto/src/main.rs b/rialto/src/main.rs
index 7b3783a..ce83624 100644
--- a/rialto/src/main.rs
+++ b/rialto/src/main.rs
@@ -31,7 +31,7 @@
 use libfdt::FdtError;
 use log::{debug, error, info};
 use vmbase::{
-    configure_global_allocator_size,
+    configure_heap,
     fdt::SwiotlbInfo,
     layout::{self, crosvm},
     main,
@@ -165,4 +165,4 @@
 }
 
 main!(main);
-configure_global_allocator_size!(SIZE_64KB);
+configure_heap!(SIZE_64KB);