diff --git a/drivers/pci/endpoint/functions/pci-epf-wrapper.h b/drivers/pci/endpoint/functions/pci-epf-wrapper.h
index 8856a979..b8eb81cd 100644
--- a/drivers/pci/endpoint/functions/pci-epf-wrapper.h
+++ b/drivers/pci/endpoint/functions/pci-epf-wrapper.h
@@ -10,6 +10,14 @@
 
 #include <linux/version.h>
 
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(5, 12, 0))
+#define lpci_epf_free_space(A, B, C)	pci_epf_free_space(A, B, C, PRIMARY_INTERFACE)
+#define lpci_epf_alloc_space(A, B, C, D)	pci_epf_alloc_space(A, B, C, D, PRIMARY_INTERFACE)
+#else
+#define lpci_epf_free_space(A, B, C)	pci_epf_free_space(A, B, C)
+#define lpci_epf_alloc_space(A, B, C, D)	pci_epf_alloc_space(A, B, C, D)
+#endif
+
 #if (LINUX_VERSION_CODE > KERNEL_VERSION(5, 14, 0))
 #define lpci_epc_write_header(A, B, C)	pci_epc_write_header(A, B, 0, C)
 #define lpci_epc_raise_irq(A, B, C, D)	pci_epc_raise_irq(A, B, 0, C, D)
@@ -17,8 +25,6 @@
 #define lpci_epc_set_msi(A, B, C)	pci_epc_set_msi(A, B, 0, C)
 #define lpci_epc_set_bar(A, B, C)	pci_epc_set_bar(A, B, 0, C)
 #define lpci_epc_unmap_addr(A, B, C)	pci_epc_unmap_addr(A, B, 0, C)
-#define lpci_epf_free_space(A, B, C)	pci_epf_free_space(A, B, C, PRIMARY_INTERFACE)
-#define lpci_epf_alloc_space(A, B, C, D)	pci_epf_alloc_space(A, B, C, D, PRIMARY_INTERFACE)
 #else
 #define lpci_epc_write_header(A, B, C)	pci_epc_write_header(A, B, C)
 #define lpci_epc_raise_irq(A, B, C, D)	pci_epc_raise_irq(A, B, C, D)
@@ -26,8 +32,6 @@
 #define lpci_epc_set_msi(A, B, C)	pci_epc_set_msi(A, B, C)
 #define lpci_epc_set_bar(A, B, C)	pci_epc_set_bar(A, B, C)
 #define lpci_epc_unmap_addr(A, B, C)	pci_epc_unmap_addr(A, B, C)
-#define lpci_epf_free_space(A, B, C)	pci_epf_free_space(A, B, C)
-#define lpci_epf_alloc_space(A, B, C, D)	pci_epf_alloc_space(A, B, C, D)
 #endif /* LINUX_VERSION_CODE */
 
 #endif /* PCI_EPF_WRAPPER_H */