diff --git a/CMakeLists.txt b/CMakeLists.txt
index 3275b9e..4541062 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -13,7 +13,7 @@ aux_source_directory(kuiper/source/op/kernels DIR_KERNEL)
 aux_source_directory(kuiper/source/sampler DIR_SAMPLE)
 
 set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/lib)
-
+find_package(GTest REQUIRED)
 add_library(llama SHARED ${DIR_TENSOR} ${DIR_BASE} ${DIR_OP} ${DIR_MODEL} ${DIR_KERNEL} ${DIR_SAMPLE})
 target_link_libraries(llama sentencepiece glog::glog  gtest gtest_main pthread)
 target_link_libraries(llama ${ARMADILLO_LIBRARIES})