diff --git a/u01/CMakeLists.txt b/u01/CMakeLists.txt
index c7232a6..4c7844b 100644
--- a/u01/CMakeLists.txt
+++ b/u01/CMakeLists.txt
@@ -5,12 +5,12 @@ add_library(tree SHARED node.cpp)
 #                ~~~~~~
 # This can be changed to STATIC to build and link a static library.
 add_executable(main main.cpp)
-target_link_libraries(main ecg_tree)
+target_link_libraries(main tree)
 
 find_package(Catch2 3)
 if(Catch2_FOUND)
   add_executable(tests tests.cpp)
-  target_link_libraries(tests PRIVATE Catch2::Catch2WithMain ecg_tree)
+  target_link_libraries(tests PRIVATE Catch2::Catch2WithMain tree)
 
   include(CTest)
   include(Catch)
@@ -18,4 +18,4 @@ if(Catch2_FOUND)
 endif()
 
 install(TARGETS main)
-install(TARGETS ecg_tree DESTINATION lib)
+install(TARGETS tree DESTINATION lib)