diff --git a/bsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c b/bsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c
index b211e27ef3ba191436dc64c8200d76c8bc5d6841..369367654b8151c651206f88d97ee20f2caeede9 100644
--- a/bsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c
+++ b/bsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c
@@ -685,7 +685,7 @@ pal_pcie_check_device_list()
   @return memory offset
 **/
 uint32_t
-pal_pcie_mem_get_offset(uint32_t bdf, uint32_t mem_type)
+pal_pcie_mem_get_offset(uint32_t bdf, PCIE_MEM_TYPE_INFO_e mem_type)
 {
 
   return 0;
diff --git a/sbsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c b/sbsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c
index 137c715811d38721c248b00bdac4c5fd55a95298..6efff10fc44a95009196d73a063f5f0df6e9f090 100644
--- a/sbsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c
+++ b/sbsa-acs-drv/files/platform/pal_linux/files/src/pal_pcie.c
@@ -646,7 +646,7 @@ pal_pcie_check_device_list()
   @return memory offset
 **/
 uint32_t
-pal_pcie_mem_get_offset(uint32_t bdf, uint32_t mem_type)
+pal_pcie_mem_get_offset(uint32_t bdf, PCIE_MEM_TYPE_INFO_e mem_type)
 {
 
   return 0;