diff --git a/pci/src/bus.rs b/pci/src/bus.rs
index 3ee9a3402..7c7fa61d1 100644
--- a/pci/src/bus.rs
+++ b/pci/src/bus.rs
@@ -124,6 +124,11 @@ impl PciBus {
         Ok(())
     }
 
+    pub fn remove_by_device(&mut self, device: &Arc<Mutex<dyn PciDevice>>) -> Result<()> {
+        self.devices.retain(|dev| !Arc::ptr_eq(dev, device));
+        Ok(())
+    }
+
     pub fn next_device_id(&self) -> u32 {
         self.devices.len() as u32
     }