diff --git a/memflow-pcileech/src/lib.rs b/memflow-pcileech/src/lib.rs
index 3f35008..161ff56 100644
--- a/memflow-pcileech/src/lib.rs
+++ b/memflow-pcileech/src/lib.rs
@@ -342,9 +342,8 @@ impl PhysicalMemory for PciLeech {
     }
 
     fn set_mem_map(&mut self, mem_map: MemoryMap<(Address, usize)>) {
-        // TODO: check if current mem_map is empty
-        // TODO: update metadata.size
-        self.mem_map = mem_map;
+        self.metadata.size = mem_map.max_address().as_usize();
+        self.mem_map.merge(mem_map)
     }
 }