diff --git a/tests/mock_tests/aclorch_ut.cpp b/tests/mock_tests/aclorch_ut.cpp
index f3ca17c8654a..302f89474734 100644
--- a/tests/mock_tests/aclorch_ut.cpp
+++ b/tests/mock_tests/aclorch_ut.cpp
@@ -23,6 +23,8 @@ extern sai_route_api_t *sai_route_api;
 extern sai_next_hop_group_api_t* sai_next_hop_group_api;
 extern string gMySwitchType;
 
+using namespace saimeta;
+
 namespace aclorch_test
 {
     using namespace std;
diff --git a/tests/mock_tests/check.h b/tests/mock_tests/check.h
index 3eddcecc2ca0..d1b095562deb 100644
--- a/tests/mock_tests/check.h
+++ b/tests/mock_tests/check.h
@@ -4,11 +4,17 @@
 #include <iostream>
 #include <vector>
 
-#include "saiattributelist.h"
+#include "SaiAttributeList.h"
+
+#include <hiredis/hiredis.h>
+#include "swss/dbconnector.h"
+#include "swss/logger.h"
+#include "sai_serialize.h"
+#include "string.h"
 
 struct Check
 {
-    static bool AttrListEq(sai_object_type_t objecttype, const std::vector<sai_attribute_t> &act_attr_list, SaiAttributeList &exp_attr_list)
+    static bool AttrListEq(sai_object_type_t objecttype, const std::vector<sai_attribute_t> &act_attr_list, saimeta::SaiAttributeList &exp_attr_list)
     {
         if (act_attr_list.size() != exp_attr_list.get_attr_count())
         {