diff --git a/core/unittest/db/test_db.cpp b/core/unittest/db/test_db.cpp
index 8d05ae9923..0a47ac1b9b 100644
--- a/core/unittest/db/test_db.cpp
+++ b/core/unittest/db/test_db.cpp
@@ -295,15 +295,6 @@ TEST_F(DBTest, SEARCH_TEST) {
         ASSERT_TRUE(stat.ok());
     }
 
-    {  // search by specify index file
-        milvus::engine::meta::DatesT dates;
-        std::vector<std::string> file_ids = {"1", "2", "3", "4", "5", "6"};
-        milvus::engine::ResultIds result_ids;
-        milvus::engine::ResultDistances result_distances;
-        stat = db_->QueryByFileID(TABLE_NAME, file_ids, k, nq, 10, xq.data(), dates, result_ids, result_distances);
-        ASSERT_TRUE(stat.ok());
-    }
-
     index.engine_type_ = (int)milvus::engine::EngineType::FAISS_IVFFLAT;
     db_->CreateIndex(TABLE_NAME, index);  // wait until build index finish
 
@@ -345,16 +336,12 @@ TEST_F(DBTest, SEARCH_TEST) {
     }
 #endif
 
-    index.engine_type_ = (int)milvus::engine::EngineType::FAISS_PQ;
-    db_->CreateIndex(TABLE_NAME, index);  // wait until build index finish
-
-    {
-        std::vector<std::string> tags;
+    {  // search by specify index file
+        milvus::engine::meta::DatesT dates;
+        std::vector<std::string> file_ids = {"1", "2", "3", "4", "5", "6"};
         milvus::engine::ResultIds result_ids;
         milvus::engine::ResultDistances result_distances;
-        stat = db_->Query(TABLE_NAME, tags, k, nq, 10, xq.data(), result_ids, result_distances);
-        ASSERT_TRUE(stat.ok());
-        stat = db_->Query(TABLE_NAME, tags, k, 1100, 10, xq.data(), result_ids, result_distances);
+        stat = db_->QueryByFileID(TABLE_NAME, file_ids, k, nq, 10, xq.data(), dates, result_ids, result_distances);
         ASSERT_TRUE(stat.ok());
     }