1 files changed, 3 insertions, 1 deletions
diff --git a/arch/x86_64/kapi/boot_modules.cpp b/arch/x86_64/kapi/boot_modules.cpp
index c86b9cf7..607aca4e 100644
--- a/arch/x86_64/kapi/boot_modules.cpp
+++ b/arch/x86_64/kapi/boot_modules.cpp
@@ -11,6 +11,7 @@
 
 #include <kstd/memory.hpp>
 #include <kstd/print.hpp>
+#include <kstd/units.hpp>
 
 #include <multiboot2/information.hpp>
 
@@ -34,7 +35,8 @@ namespace kapi::boot_modules
         .name = module.string(),
         .start_address =
             memory::linear_address{module.start_address + std::bit_cast<std::uintptr_t>(&arch::boot::TEACHOS_VMA)},
-        .size = module.end_address - module.start_address};
+        .size = kstd::bytes{module.end_address - module.start_address},
+      };
       auto device = kstd::make_shared<kapi::boot_modules::device>(index, boot_module);
       bus.add_child(device);
     });