diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 03715a2..d370e0e 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -2,5 +2,5 @@ add_executable(n-queens
     ./n-queens.cpp
 )
 
-target_include_directories(proxy-network PUBLIC ${CMAKE_CURRENT_LIST_DIR})
+target_include_directories(n-queens PUBLIC ${CMAKE_CURRENT_LIST_DIR})
 
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index bb1f7c7..72a93d2 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -1,5 +1,5 @@
-add_subdirectory(Unity)
-include_directories("${CMAKE_CURRENT_SOURCE_DIR}")
+find_package(unity REQUIRED)
+include_directories(${CMAKE_SOURCE_DIR}/src)
 add_executable(testing 
     ./test.cpp
 )