diff --git a/src/include/device/pci.h b/src/include/device/pci.h
index 8175970..0670da4 100644
--- a/src/include/device/pci.h
+++ b/src/include/device/pci.h
@@ -102,8 +102,14 @@
 
 #endif /* ! __SIMPLE_DEVICE__ */
 
+#ifdef __PRE_RAM__
+unsigned pci_find_next_capability(pci_devfn_t dev, unsigned cap, unsigned last);
+unsigned pci_find_capability(pci_devfn_t dev, unsigned cap);
+#else /* !__PRE_RAM__ */
 unsigned pci_find_next_capability(device_t dev, unsigned cap, unsigned last);
 unsigned pci_find_capability(device_t dev, unsigned cap);
+#endif /* __PRE_RAM__ */
+
 void pci_early_bridge_init(void);
 int pci_early_device_probe(u8 bus, u8 dev, u32 mmio_base);