diff --git a/CMakeLists.txt b/CMakeLists.txt
index bbd066c..748c149 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -4,7 +4,9 @@ include("test/lib/KUnity/dev-utility/BuildKernelModule.cmake")
 SET(module_name kunity_test_runner_module)
 project (${module_name})
 
+if(NOT KERNEL_DIR)
 Set(KERNEL_DIR "/lib/modules/${CMAKE_SYSTEM_VERSION}/build" )
+endif()
 set(CMAKE_C_FLAGS  "${CMAKE_CXX_FLAGS} -Wall -Wextra -Werror -pedantic -Weverything")
 
 file(GLOB module_src_files
diff --git a/test/lib/KUnity b/test/lib/KUnity
index 2909c6a..ec3f646 160000
--- a/test/lib/KUnity
+++ b/test/lib/KUnity
@@ -1 +1 @@
-Subproject commit 2909c6a09114dcbc5056c897e3b567814d83b370
+Subproject commit ec3f646c5baf1d74e6bc3de24bb31cbbd368f99f