diff --git a/core/unittest/server/test_config.cpp b/core/unittest/server/test_config.cpp
index cee17542b5f99bbc3495d181936de55f8f0238d6..3c5f0acffef435d4c004e5dff432a527a12ddf70 100644
--- a/core/unittest/server/test_config.cpp
+++ b/core/unittest/server/test_config.cpp
@@ -18,11 +18,13 @@
 #include <gtest/gtest.h>
 #include <gtest/gtest-death-test.h>
 
-#include "config/ConfigMgr.h"
+#include "config/YamlConfigMgr.h"
 #include "utils/CommonUtil.h"
 #include "utils/ValidationUtil.h"
 #include "server/Config.h"
 
+using namespace milvus;
+
 namespace {
 
 static const char *CONFIG_FILE_PATH = "./milvus/conf/server_config.yaml";
@@ -35,16 +37,16 @@ static constexpr uint64_t GB = MB * 1024;
 } // namespace
 
 TEST(ConfigTest, CONFIG_TEST) {
-    milvus::server::ConfigMgr *config_mgr = milvus::server::ConfigMgr::GetInstance();
+    server::ConfigMgr *config_mgr = server::YamlConfigMgr::GetInstance();
 
-    milvus::ErrorCode err = config_mgr->LoadConfigFile("");
-    ASSERT_EQ(err, milvus::SERVER_UNEXPECTED_ERROR);
+    milvus::Status s = config_mgr->LoadConfigFile("");
+    ASSERT_FALSE(s.ok());
 
-    err = config_mgr->LoadConfigFile(LOG_FILE_PATH);
-    ASSERT_EQ(err, milvus::SERVER_UNEXPECTED_ERROR);
+    s = config_mgr->LoadConfigFile(LOG_FILE_PATH);
+    ASSERT_FALSE(s.ok());
 
-    err = config_mgr->LoadConfigFile(CONFIG_FILE_PATH);
-    ASSERT_EQ(err, milvus::SERVER_SUCCESS);
+    s = config_mgr->LoadConfigFile(CONFIG_FILE_PATH);
+    ASSERT_TRUE(s.ok());
 
     config_mgr->Print();