diff --git a/CMakeLists.txt b/CMakeLists.txt
index 07c43fbc3..a93ee08c0 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -236,11 +236,11 @@ message(STATUS "Add unit test binaries: ${UNIT_TESTS}")
 # -----------------------------------------------------------------------------
 
 # includes
-include_directories(${CMAKE_SOURCE_DIR}/include)
 include_directories(${src_dir})
 include_directories(${build_include_dir})
 include_directories(${fmt_source_dir})
 include_directories(${yaml_include_dir})
+include_directories(${GTEST_INCLUDE_DIRS} ${GMOCK_INCLUDE_DIRS})
 # -----------------------------------------------------------------------------
 
 # openCypher parser -----------------------------------------------------------
diff --git a/tests/unit/CMakeLists.txt b/tests/unit/CMakeLists.txt
index 191ebc0e3..6c1aec109 100644
--- a/tests/unit/CMakeLists.txt
+++ b/tests/unit/CMakeLists.txt
@@ -27,7 +27,7 @@ foreach(test_cpp ${test_type_cpps})
 
     # link libraries
     # gtest
-    target_link_libraries(${target_name} gtest gtest_main)
+    target_link_libraries(${target_name} gtest gtest_main gmock)
     # filesystem
     target_link_libraries(${target_name} stdc++fs)
     # threads (cross-platform)