diff --git a/unit_test/CMakeLists.txt b/unit_test/CMakeLists.txt
index 24de945..5e4cd78 100644
--- a/unit_test/CMakeLists.txt
+++ b/unit_test/CMakeLists.txt
@@ -37,12 +37,6 @@ AUX_SOURCE_DIRECTORY(./test TEST_SRC)
 INCLUDE_DIRECTORIES(${CMAKE_SOURCE_DIR}/include)
 
 # Now simply link against gtest or gtest_main as needed. Eg
-add_executable(ztp_unit ${TEST_SRC})
-
-#if(ZTP_LIBRARY_FOUND)
-target_link_libraries(ztp_unit ${ZTP_LIBRARY} ztp gtest_main)
-#else()
-#target_link_libraries(ztp_unit gtest_main ztp)
-#endif()
-
-add_test(NAME ztp_unit_test COMMAND ztp_unit)
\ No newline at end of file
+ADD_EXECUTABLE(ztp_unit ${TEST_SRC})
+TARGET_LINK_LIBRARIES(ztp_unit ${ZTP_LIBRARY} ztp gtest_main)
+ADD_TEST(NAME ztp_unit_test COMMAND ztp_unit)
\ No newline at end of file
diff --git a/unit_test/test/json_test.cpp b/unit_test/test/json_test.cpp
index 0c532fc..522fe2c 100644
--- a/unit_test/test/json_test.cpp
+++ b/unit_test/test/json_test.cpp
@@ -2,10 +2,11 @@
 // Created by xajhu on 2019/11/29 0029.
 //
 
-#include "gtest/gtest.h"
 #include <iostream>
 using namespace std;
 
+#include "gtest/gtest.h"
+
 extern "C" {
 #include "json_interface.h"
 }
@@ -28,7 +29,7 @@ TEST(json_encode_test, retOK)
     ASSERT_EQ(0,ret);
 }
 
-int main(int argc, char** argv)
+auto main(int argc, char* argv[]) -> int
 {
     int ret;