diff --git a/shared/test/unit_test/xe_hpc_core/test_device_caps_xe_hpc_core.cpp b/shared/test/unit_test/xe_hpc_core/test_device_caps_xe_hpc_core.cpp
index 57a89f75ff..2307d81bae 100644
--- a/shared/test/unit_test/xe_hpc_core/test_device_caps_xe_hpc_core.cpp
+++ b/shared/test/unit_test/xe_hpc_core/test_device_caps_xe_hpc_core.cpp
@@ -11,6 +11,8 @@
 #include "shared/test/common/helpers/hw_helper_tests.h"
 #include "shared/test/common/test_macros/header/per_product_test_definitions.h"
 
+#include "hw_cmds_xe_hpc_core_base.h"
+
 using namespace NEO;
 
 using XeHpcCoreDeviceCaps = Test<DeviceFixture>;
diff --git a/shared/test/unit_test/xe_hpg_core/test_device_caps_xe_hpg_core.cpp b/shared/test/unit_test/xe_hpg_core/test_device_caps_xe_hpg_core.cpp
index b5e62776d3..f75dd7331c 100644
--- a/shared/test/unit_test/xe_hpg_core/test_device_caps_xe_hpg_core.cpp
+++ b/shared/test/unit_test/xe_hpg_core/test_device_caps_xe_hpg_core.cpp
@@ -10,6 +10,8 @@
 #include "shared/test/common/helpers/hw_helper_tests.h"
 #include "shared/test/common/test_macros/header/per_product_test_definitions.h"
 
+#include "hw_cmds_xe_hpg_core_base.h"
+
 using namespace NEO;
 
 using XeHpgCoreDeviceCaps = Test<DeviceFixture>;