1 files changed, 8 insertions, 7 deletions
diff --git a/kernel/kernel/devices/block_device.tests.cpp b/kernel/kernel/devices/block_device.tests.cpp
index 98722638..23af74ef 100644
--- a/kernel/kernel/devices/block_device.tests.cpp
+++ b/kernel/kernel/devices/block_device.tests.cpp
@@ -5,20 +5,21 @@
 #include <kstd/memory.hpp>
 #include <kstd/print.hpp>
 #include <kstd/string.hpp>
+#include <kstd/units.hpp>
 #include <kstd/vector.hpp>
 
 #include <catch2/catch_test_macros.hpp>
 
-#include <cstddef>
+using namespace kstd::units_literals;
 
 SCENARIO("Block device construction", "[devices][block_device]")
 {
   GIVEN("parameters for a block device")
   {
-    size_t major = 1;
-    size_t minor = 0;
-    kstd::string name = "test_block_device";
-    size_t block_size = 512;
+    auto major = 1uz;
+    auto minor = 0uz;
+    auto name = "test_block_device";
+    auto block_size = 512_B;
 
     WHEN("constructing a block device")
     {
@@ -31,7 +32,7 @@ SCENARIO("Block device construction", "[devices][block_device]")
         REQUIRE(device->minor() == minor);
         REQUIRE(device->name() == name);
         REQUIRE(device->block_size() == block_size);
-        REQUIRE(device->capacity() == 3 * 512);
+        REQUIRE(device->capacity() == 3 * 512_B);
       }
     }
 
@@ -39,7 +40,7 @@ SCENARIO("Block device construction", "[devices][block_device]")
     {
       THEN("the constructor panics")
       {
-        REQUIRE_THROWS_AS((kernel::tests::devices::block_device(major, minor, name, 0)), kernel::tests::cpu::halt);
+        REQUIRE_THROWS_AS((kernel::tests::devices::block_device(major, minor, name, 0_B)), kernel::tests::cpu::halt);
       }
     }
   }