diff --git a/src/include/device/pci_ops.h b/src/include/device/pci_ops.h
index 7fe7d42..a167825 100644
--- a/src/include/device/pci_ops.h
+++ b/src/include/device/pci_ops.h
@@ -215,6 +215,6 @@
  *
  * Returns true if PCI device is wake source, false otherwise.
  */
-bool pci_dev_is_wake_source(const struct device *dev);
+bool pci_dev_is_wake_source(pci_devfn_t dev);
 
 #endif /* PCI_OPS_H */