diff --git a/test_pool/pcie/operating_system/test_p001.c b/test_pool/pcie/operating_system/test_p001.c
index 3d88cfca..d17aa939 100644
--- a/test_pool/pcie/operating_system/test_p001.c
+++ b/test_pool/pcie/operating_system/test_p001.c
@@ -16,6 +16,7 @@
  **/
 #include "val/common/include/acs_val.h"
 #include "val/common/include/acs_pcie.h"
+#include "val/sbsa/include/sbsa_acs_pcie.h"
 
 #define TEST_NUM   (ACS_PCIE_TEST_NUM_BASE + 1)
 #define TEST_DESC  "Check ECAM Presence               "
diff --git a/test_pool/pcie/operating_system/test_p005.c b/test_pool/pcie/operating_system/test_p005.c
index cb673b51..b45c1ca3 100644
--- a/test_pool/pcie/operating_system/test_p005.c
+++ b/test_pool/pcie/operating_system/test_p005.c
@@ -18,6 +18,7 @@
 #include "val/common/include/acs_pe.h"
 #include "val/common/include/acs_pcie.h"
 #include "val/common/include/acs_memory.h"
+#include "val/sbsa/include/sbsa_acs_pcie.h"
 
 /* SBSA-checklist 63 & 64 */
 #define TEST_NUM   (ACS_PCIE_TEST_NUM_BASE + 5)
diff --git a/test_pool/pcie/operating_system/test_p009.c b/test_pool/pcie/operating_system/test_p009.c
index 762dc745..c706fd4a 100644
--- a/test_pool/pcie/operating_system/test_p009.c
+++ b/test_pool/pcie/operating_system/test_p009.c
@@ -18,6 +18,7 @@
 #include "val/common/include/acs_val.h"
 #include "val/common/include/acs_pcie.h"
 #include "val/common/include/acs_memory.h"
+#include "val/sbsa/include/sbsa_acs_pcie.h"
 
 #define TEST_NUM   (ACS_PCIE_TEST_NUM_BASE + 9)
 #define TEST_DESC  "Check all MSI(X) vectors are LPIs "
diff --git a/test_pool/smmu/operating_system/test_i016.c b/test_pool/smmu/operating_system/test_i016.c
index fc7180b9..042f777b 100644
--- a/test_pool/smmu/operating_system/test_i016.c
+++ b/test_pool/smmu/operating_system/test_i016.c
@@ -22,6 +22,7 @@
 #include "val/sbsa/include/sbsa_acs_iovirt.h"
 #include "val/common/include/acs_smmu.h"
 #include "val/common/include/acs_memory.h"
+#include "val/sbsa/include/sbsa_acs_smmu.h"
 
 #define TEST_NUM   (ACS_SMMU_TEST_NUM_BASE + 16)
 #define TEST_RULE  "S_L7SM_02"