1 files changed, 2 insertions, 2 deletions
diff --git a/src/block_device/memory_disk.rs b/src/block_device/memory_disk.rs
index 3876736..0639d3e 100644
--- a/src/block_device/memory_disk.rs
+++ b/src/block_device/memory_disk.rs
@@ -9,9 +9,9 @@ pub struct MemoryDisk {
 }
 
 impl MemoryDisk {
-    pub fn new() -> Self {
+    pub fn new(block_number: usize) -> Self {
         Self {
-            arena: RefCell::new(vec![0u8; BLOCK_SIZE * 16384]),
+            arena: RefCell::new(vec![0u8; BLOCK_SIZE * block_number]),
         }
     }
 }