diff --git a/imperative/test/CMakeLists.txt b/imperative/test/CMakeLists.txt
index 5599b6ca..280da981 100644
--- a/imperative/test/CMakeLists.txt
+++ b/imperative/test/CMakeLists.txt
@@ -18,7 +18,7 @@ target_include_directories(imperative_test PRIVATE ${MODULE_SRC_INCLUDE} ${PYTHO
 target_compile_definitions(imperative_test PRIVATE MODULE_NAME=C)
 target_compile_options(imperative_test PRIVATE -Wno-unused-parameter)
 
-set(LINK_LIBS megbrain megdnn gtest pybind11::embed)
+set(LINK_LIBS megbrain megdnn gtest gmock pybind11::embed)
 if(MGE_WITH_CUDA)
     list(APPEND LINK_LIBS cudart)
 endif()
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 4c76dfbf..ab94fe1b 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -17,7 +17,7 @@ endif()
 
 
 add_executable(megbrain_test ${SOURCES})
-target_link_libraries(megbrain_test gtest)
+target_link_libraries(megbrain_test gtest gmock)
 if(MSVC OR WIN32)
     target_link_libraries(megbrain_test megbrain megdnn)
 else()
diff --git a/test/src/main.cpp b/test/src/main.cpp
index e36d670f..be296b14 100644
--- a/test/src/main.cpp
+++ b/test/src/main.cpp
@@ -13,6 +13,7 @@
 #include "megbrain/test/helper.h"
 
 #include <gtest/gtest.h>
+#include <gmock/gmock.h>
 #include <cstdlib>
 
 extern "C" int gtest_main(int argc, char** argv) {
@@ -32,7 +33,7 @@ extern "C" int gtest_main(int argc, char** argv) {
     auto&& listeners = ::testing::UnitTest::GetInstance()->listeners();
     MGB_TRY {
         srand(time(nullptr));
-        ::testing::InitGoogleTest(&argc, argv);
+        ::testing::InitGoogleMock(&argc, argv);
         listeners.Append(&mgb::RNGSeedManager::inst());
         auto rst = RUN_ALL_TESTS();
         mgb::CompNode::finalize();