diff --git a/cmake/util.cmake b/cmake/util.cmake
index 8a71b23c62d9fd79ffeb6b1b2281e0f2728db5a8..0f60d9c6a31e63caf44eb19806bb162e4093a29b 100644
--- a/cmake/util.cmake
+++ b/cmake/util.cmake
@@ -107,7 +107,6 @@ function(link_paddle_exe TARGET_NAME)
         paddle_parameter
         paddle_proto
         paddle_cuda
-        paddle_test_main
         ${METRIC_LIBS}
         ${PROTOBUF_LIBRARY}
         ${LIBGLOG_LIBRARY}
@@ -156,7 +155,7 @@ endfunction()
 function(link_paddle_test TARGET_NAME)
     link_paddle_exe(${TARGET_NAME})
     target_link_libraries(${TARGET_NAME} ${GTEST_MAIN_LIBRARIES}
-        ${GTEST_LIBRARIES})
+        ${GTEST_LIBRARIES} paddle_test_main)
 endfunction()
 
 # add_unittest_without_exec